Browse Source

添加保存模型权重文件的间隔

233
fhw 1 year ago
parent
commit
7dc7263d39
  1. 2
      configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py

2
configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py

@ -4,6 +4,8 @@ _base_ = [
]
evaluation = dict(interval=1, metric='mAP', save_best='AP')
checkpoint_config = dict(interval=1) # 保存模型权重文件的间隔
optimizer = dict(type='AdamW',
lr=5e-4, betas=(0.9, 0.999), weight_decay=0.1,
constructor='LayerDecayOptimizerConstructor',

Loading…
Cancel
Save