add random seed in train (#358)

2020-11-03 10:08:30 +08:00 · 2020-11-03 10:08:30 +08:00 · 14063829db
parent 7bb8f1c2ef
commit 14063829db
1 changed files with 2 additions and 0 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -52,6 +52,8 @@ def parse_args():


 def main(args):
+    paddle.seed(123)
+
    config = get_config(args.config, overrides=args.override, show=True)
    # assign the place
    use_gpu = config.get("use_gpu", True)