Skip to content

Commit cc1b8bc

Browse files
committed
Update param_parser.py
1 parent b5e7983 commit cc1b8bc

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎utils/param_parser.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def parameter_parser():
118118

119119
parser.add_argument("--batch-size",
120120
type=int,
121-
default=256,
121+
default=64,
122122
help="Batch size. (default: 256)")
123123

124124
parser.add_argument("--learning-rate",
@@ -138,7 +138,7 @@ def parameter_parser():
138138

139139
parser.add_argument("--evaluate-steps",
140140
type=int,
141-
default=500,
141+
default=50,
142142
help="Evaluate model on val set after how many steps. (default: 500)")
143143

144144
parser.add_argument("--norm-ratio",
@@ -153,7 +153,7 @@ def parameter_parser():
153153

154154
parser.add_argument("--checkpoint-steps",
155155
type=int,
156-
default=500,
156+
default=50,
157157
help="Save model after how many steps. (default: 500)")
158158

159159
parser.add_argument("--num-checkpoints",

0 commit comments

Comments
 (0)