Skip to content

Commit f7cd3c9

Browse files
committed
Bump up the base LR
1 parent e9278ed commit f7cd3c9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
TRAINING_PARAMS = {
2626
"mnasnet0_5": {
2727
"num_epochs": 350,
28-
"base_lr": 1.0,
28+
"base_lr": 2.0,
2929
"momentum": 0.9,
3030
"weight_decay": 0,
3131
"batch_size": 1000,

0 commit comments

Comments
 (0)