恢复到10.18-tm2-1的config和py, 12.8-ST_ST-basic2

Tianrui-Li · Dec 9, 2023 · f45207f · f45207f
1 parent 8649d9d
commit f45207f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/transformer/ST_ST.py b/configs/transformer/ST_ST.py
@@ -77,7 +77,7 @@
     dict(type='ToTensor', keys=['keypoint'])
 ]
 data = dict(
-    videos_per_gpu=2,
+    videos_per_gpu=32,
     workers_per_gpu=8,
     test_dataloader=dict(videos_per_gpu=8),
     train=dict(type=dataset_type, ann_file=ann_file, pipeline=train_pipeline, split='xsub_train'),