batch_size=6

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

@@ -148,10 +148,10 @@ val_pipeline = [
 test_pipeline = val_pipeline
 
 data = dict(
-    samples_per_gpu=12,
+    samples_per_gpu=6,
     workers_per_gpu=4,
-    val_dataloader=dict(samples_per_gpu=12),
+    val_dataloader=dict(samples_per_gpu=6),
-    test_dataloader=dict(samples_per_gpu=12),
+    test_dataloader=dict(samples_per_gpu=6),
     train=dict(
         type='TopDownCocoDataset',
         ann_file=f'{data_root}/annotations/person_keypoints_train2017.json',