From 1be53eab44a569519fb618331034472fee014690 Mon Sep 17 00:00:00 2001
From: fhw <fhw@qq.com>
Date: Wed, 10 Jul 2024 14:56:13 +0800
Subject: [PATCH] batch_size=6

---
 .../topdown_heatmap/coco/ViTSam_base_coco_256x192.py        | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git 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
index 1f8227e..d47e16c 100644
--- 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
@@ -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),
-    test_dataloader=dict(samples_per_gpu=12),
+    val_dataloader=dict(samples_per_gpu=6),
+    test_dataloader=dict(samples_per_gpu=6),
     train=dict(
         type='TopDownCocoDataset',
         ann_file=f'{data_root}/annotations/person_keypoints_train2017.json',