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

fix hyperparams & mv to PULC

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