21 lines
272 B
YAML
21 lines
272 B
YAML
INPUT:
|
|
PIXEL_MEAN: [0., 0., 0.]
|
|
PIXEL_STD: [1., 1., 1.]
|
|
|
|
DATALOADER:
|
|
TRAIN_X:
|
|
BATCH_SIZE: 16
|
|
TEST:
|
|
BATCH_SIZE: 16
|
|
|
|
OPTIM:
|
|
NAME: "sgd"
|
|
LR: 0.0005
|
|
STEPSIZE: [20]
|
|
MAX_EPOCH: 25
|
|
|
|
TRAINER:
|
|
DDAIG:
|
|
G_ARCH: "fcn_3x64_gctx"
|
|
WARMUP: 3
|
|
LMDA: 0.3 |