提交 e31509ee 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix hyperparams & mv to PULC

上级 19840cb0
......@@ -7,7 +7,7 @@ Global:
save_interval: 10
eval_during_train: True
eval_interval: 10
epochs: 20
epochs: 60
print_batch_step: 10
use_visualdl: False
# used for static mode and model export
......@@ -20,6 +20,7 @@ Arch:
pretrained: True
use_ssld: True
class_num: 4
use_last_conv: False
# loss function config for traing/eval process
Loss:
......@@ -37,7 +38,7 @@ Optimizer:
momentum: 0.9
lr:
name: Cosine
learning_rate: 0.56
learning_rate: 0.14
warmup_epoch: 5
regularizer:
name: 'L2'
......@@ -79,11 +80,11 @@ DataLoader:
- RandomRot90:
sampler:
name: DistributedBatchSampler
batch_size: 512
batch_size: 256
drop_last: False
shuffle: True
loader:
num_workers: 16
num_workers: 12
use_shared_memory: True
Eval:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册