From 2576a411a5c1c0b49aa6638a682165d1e92aed33 Mon Sep 17 00:00:00 2001 From: iperov Date: Wed, 21 Nov 2018 16:29:48 +0400 Subject: [PATCH] Update main.py --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 9122cc5..7ca8fd7 100644 --- a/main.py +++ b/main.py @@ -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 (