forked from lnsmith54/super-convergence
-
Notifications
You must be signed in to change notification settings - Fork 0
/
clrsolver.prototxt
26 lines (26 loc) · 954 Bytes
/
clrsolver.prototxt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
net: "examples/sc/architectures/arch.prototxt"
#net: "examples/sc/architectures/Resnet56Cifar.prototxt"
#net: "examples/sc/architectures/Resnet56Dropout.prototxt"
#net: "examples/sc/architectures/Resnet56Cifar-mod.prototxt"
#net: "examples/sc/architectures/Resnet56Cifar_changeBatchNorm.prototxt"
#net: "examples/sc/architectures/Resnet56Cifar_changePooling.prototxt"
#net: "examples/sc/architectures/bottleneckResnet56.prototxt"
#net: "examples/sc/architectures/Resnet20Cifar.prototxt"
#net: "examples/sc/architectures/resnet20bn.prototxt"
#net: "examples/sc/architectures/ResNeXt56.prototxt"
#net: "examples/sc/architectures/Resnet20Cifar.prototxt"
#
test_iter: 200
test_interval: 100
display: 100
lr_policy: "triangular"
base_lr: 0.1
max_lr: 3.0
stepsize: 5000
max_iter: 10000
snapshot: 20000
snapshot_prefix: "examples/sc/snapshots/fig1a"
solver_mode: GPU
weight_decay: 1e-4
momentum: 0.9
#type: "Nesterov" # "RMSProp" # "Adam" # "AdaGrad" #