model_name: ResNet2 | |
model_hparams: | |
num_classes: 10 | |
c_hidden: | |
- 16 | |
- 32 | |
- 64 | |
num_blocks: | |
- 3 | |
- 3 | |
- 3 | |
act_fn_name: relu | |
optimizer_name: SGD | |
optimizer_hparams: | |
lr: 0.1 | |
momentum: 0.9 | |
weight_decay: 0.0001 | |
model_name: ResNet2 | |
model_hparams: | |
num_classes: 10 | |
c_hidden: | |
- 16 | |
- 32 | |
- 64 | |
num_blocks: | |
- 3 | |
- 3 | |
- 3 | |
act_fn_name: relu | |
optimizer_name: SGD | |
optimizer_hparams: | |
lr: 0.1 | |
momentum: 0.9 | |
weight_decay: 0.0001 | |