From 275c5a2ba93d0791e7b8009e3a595f4ac7554989 Mon Sep 17 00:00:00 2001
From: Topdu <784990967@qq.com>
Date: Mon, 30 May 2022 08:38:07 +0000
Subject: [PATCH] modify vitstr config batchsize

---
 configs/rec/rec_vitstr_none_ce.yml                          | 2 +-
 test_tipc/configs/rec_vitstr_none_ce/rec_vitstr_none_ce.yml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/rec/rec_vitstr_none_ce.yml b/configs/rec/rec_vitstr_none_ce.yml
index 065cc785..c71eba0c 100644
--- 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:
diff --git 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
index ccd93c9e..4e8151f5 100644
--- 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:
-- 
GitLab