15 lines
183 B
YAML
15 lines
183 B
YAML
DATALOADER:
|
|
TRAIN_X:
|
|
BATCH_SIZE: 32
|
|
TEST:
|
|
BATCH_SIZE: 32
|
|
|
|
OPTIM:
|
|
NAME: "sgd"
|
|
LR: 0.0001
|
|
STEPSIZE: [2]
|
|
MAX_EPOCH: 2
|
|
|
|
TRAIN:
|
|
PRINT_FREQ: 50
|
|
COUNT_ITER: "train_u" |