Skip to content

Commit 0240e34

Browse files
committed
add mobilenetv1 yml
1 parent b1d48dc commit 0240e34

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

yamls/mobilenetv1.yml

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
CLASSIFIER: mobilenet_v1
2+
YAML: common.yml
3+
COMMON:
4+
workers: 4
5+
batch_size: 128
6+
weight_decay: 5e-4
7+
epochs: 300
8+
data_path: data/cifar-100-python.tar.gz
9+
logdir: logs
10+
verbose: 1
11+
seed: 2021
12+
OPTIMIZER:
13+
name: momentum
14+
params:
15+
momentum: 0.9
16+
SCHEDULER:
17+
name: cosine
18+
warm_up_step: 2000

0 commit comments

Comments
 (0)