parent
ac466b68da
commit
f02a4630b2
|
@ -4,12 +4,12 @@ Global:
|
||||||
pretrained_model: null #"metabin_resnet50_final"
|
pretrained_model: null #"metabin_resnet50_final"
|
||||||
output_dir: "./output/"
|
output_dir: "./output/"
|
||||||
device: "gpu"
|
device: "gpu"
|
||||||
iter_per_epoch: 145
|
iter_per_epoch: &iter_per_epoch 50
|
||||||
warmup_iter: 10
|
warmup_iter: 10
|
||||||
save_interval: 40
|
save_interval: 40
|
||||||
eval_during_train: True
|
eval_during_train: True
|
||||||
eval_interval: 10
|
eval_interval: 8
|
||||||
epochs: &epochs 120
|
epochs: &epochs 348 # 348*50 = 120*145 = 17400 iters
|
||||||
print_batch_step: 20
|
print_batch_step: 20
|
||||||
use_visualdl: False
|
use_visualdl: False
|
||||||
eval_mode: "retrieval"
|
eval_mode: "retrieval"
|
||||||
|
@ -238,10 +238,10 @@ Optimizer:
|
||||||
name: MultiStepDecay
|
name: MultiStepDecay
|
||||||
epochs: *epochs
|
epochs: *epochs
|
||||||
learning_rate: 0.01
|
learning_rate: 0.01
|
||||||
step_each_epoch: 145
|
step_each_epoch: *iter_per_epoch
|
||||||
milestones: [50, 90]
|
milestones: [145, 261]
|
||||||
gamma: 0.1
|
gamma: 0.1
|
||||||
warmup_epoch: 10
|
warmup_epoch: 29
|
||||||
warmup_start_lr: 0.0001
|
warmup_start_lr: 0.0001
|
||||||
by_epoch: False
|
by_epoch: False
|
||||||
last_epoch: -1
|
last_epoch: -1
|
||||||
|
@ -260,7 +260,7 @@ Optimizer:
|
||||||
lr:
|
lr:
|
||||||
name: Cyclic
|
name: Cyclic
|
||||||
epochs: *epochs
|
epochs: *epochs
|
||||||
step_each_epoch: 145
|
step_each_epoch: *iter_per_epoch
|
||||||
base_learning_rate: 0.001
|
base_learning_rate: 0.001
|
||||||
max_learning_rate: 0.1
|
max_learning_rate: 0.1
|
||||||
warmup_epoch: 0
|
warmup_epoch: 0
|
||||||
|
|
Loading…
Reference in New Issue