12 lines
161 B
YAML
12 lines
161 B
YAML
DATALOADER:
|
|
TRAIN_X:
|
|
BATCH_SIZE: 64
|
|
TEST:
|
|
BATCH_SIZE: 100
|
|
NUM_WORKERS: 8
|
|
|
|
OPTIM:
|
|
NAME: "sgd"
|
|
LR: 0.001
|
|
MAX_EPOCH: 50
|
|
LR_SCHEDULER: "cosine" |