modify vitstr config batchsize

275c5a2b · Topdu · c269e1e4 · 275c5a2b · 275c5a2b
Showing with 2 addition and 2 deletion

configs/rec/rec_vitstr_none_ce.yml configs/rec/rec_vitstr_none_ce.yml +1 -1

test_tipc/configs/rec_vitstr_none_ce/rec_vitstr_none_ce.yml test_tipc/configs/rec_vitstr_none_ce/rec_vitstr_none_ce.yml +1 -1

未找到文件。
--- a/configs/rec/rec_vitstr_none_ce.yml
+++ b/configs/rec/rec_vitstr_none_ce.yml
@@ -75,7 +75,7 @@ Train:
    shuffle: True
    batch_size_per_card: 48
    drop_last: True
-    num_workers: 2
+    num_workers: 8
 Eval:
  dataset:

--- a/test_tipc/configs/rec_vitstr_none_ce/rec_vitstr_none_ce.yml
+++ b/test_tipc/configs/rec_vitstr_none_ce/rec_vitstr_none_ce.yml
@@ -75,7 +75,7 @@ Train:
    shuffle: True
    batch_size_per_card: 48
    drop_last: True
-    num_workers: 2
+    num_workers: 8
 Eval:
  dataset: