There was an error while loading. Please reload this page.
1 parent b5e7983 commit cc1b8bcCopy full SHA for cc1b8bc
1 file changed
utils/param_parser.py
@@ -118,7 +118,7 @@ def parameter_parser():
118
119
parser.add_argument("--batch-size",
120
type=int,
121
- default=256,
+ default=64,
122
help="Batch size. (default: 256)")
123
124
parser.add_argument("--learning-rate",
@@ -138,7 +138,7 @@ def parameter_parser():
138
139
parser.add_argument("--evaluate-steps",
140
141
- default=500,
+ default=50,
142
help="Evaluate model on val set after how many steps. (default: 500)")
143
144
parser.add_argument("--norm-ratio",
@@ -153,7 +153,7 @@ def parameter_parser():
153
154
parser.add_argument("--checkpoint-steps",
155
156
157
help="Save model after how many steps. (default: 500)")
158
159
parser.add_argument("--num-checkpoints",
0 commit comments