--check-step, --batch-grow 옵션 삭제 #30

kakao · Feb 10, 2019 · 90bd910 · 90bd910
1 parent 2b24c89
commit 90bd910
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/train/train.py b/train/train.py
@@ -28,8 +28,7 @@ def run(args: Namespace):
     Args:
         args:  program arguments (config)
     """
-    trner = Trainer(args)
-    trner.train()
+    Trainer(args).train()
 
 
 ########
@@ -59,10 +58,6 @@ def main():
                         type=float, default=0.9)
     parser.add_argument('--batch-size', help='batch size <default: 500>', metavar='INT', type=int,
                         default=500)
-    parser.add_argument('--check-step', help='check every N step <default: 10000>', metavar='INT',
-                        type=int, default=10000)
-    parser.add_argument('--batch-grow', help='grow batch size by 1 per N step <default: 10000>',
-                        metavar='INT', type=int, default=10000)
     parser.add_argument('--patience', help='maximum patience count to revert model <default: 10>',
                         metavar='INT', type=int, default=10)
     parser.add_argument('--gpu-num', help='GPU number to use <default: 0>', metavar='INT', type=int,