Update main.py

2018-11-21 16:29:48 +04:00
parent ef7fdd49a0
commit 2576a411a5
1 changed files with 1 additions and 1 deletions
@@ -68,7 +68,7 @@ if __name__ == "__main__":
            arguments.target_epoch = int ( os.environ['DFL_TARGET_EPOCH'] )
    
        if 'DFL_BATCH_SIZE' in os.environ.keys():
-            arguments.batch_size = int ( os.environ['DFL_TARGET_EPOCH'] )
+            arguments.batch_size = int ( os.environ['DFL_BATCH_SIZE'] )

        from mainscripts import Trainer
        Trainer.main (