更改训练轮数

1 year ago · 33ef51cf1a
1 changed files with 1 additions and 1 deletions
--- a/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py
+++ b/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py
@ -30,7 +30,7 @@ lr_config = dict(
    warmup_iters=500,
    warmup_ratio=0.001,
    step=[170, 200])
-total_epochs = 210
+total_epochs = 20
 target_type = 'GaussianHeatmap'
 channel_cfg = dict(
    num_output_channels=17,