perf: 优化io配置

MaaAssistantArknights · Dec 11, 2022 · e43d5d3 · e43d5d3
1 parent 85ea0b6
commit e43d5d3
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/Vision/OCR/ch_PP-OCRv3_rec_distillation.yml b/Vision/OCR/ch_PP-OCRv3_rec_distillation.yml
@@ -176,7 +176,7 @@ Train:
     shuffle: true
     batch_size_per_card: 128
     drop_last: true
-    num_workers: 1
+    num_workers: 4
 Eval:
   dataset:
     name: SimpleDataSet
@@ -201,4 +201,4 @@ Eval:
     shuffle: false
     drop_last: false
     batch_size_per_card: 128
-    num_workers: 1
+    num_workers: 4
diff --git a/Vision/OCR/chinese_cht_PP-OCRv3_rec.yml b/Vision/OCR/chinese_cht_PP-OCRv3_rec.yml
@@ -101,7 +101,7 @@ Train:
     shuffle: true
     batch_size_per_card: 128
     drop_last: true
-    num_workers: 1
+    num_workers: 4
 Eval:
   dataset:
     name: SimpleDataSet
@@ -126,4 +126,4 @@ Eval:
     shuffle: false
     drop_last: false
     batch_size_per_card: 128
-    num_workers: 1
+    num_workers: 4
diff --git a/Vision/OCR/japan_PP-OCRv3_rec.yml b/Vision/OCR/japan_PP-OCRv3_rec.yml
@@ -101,7 +101,7 @@ Train:
     shuffle: true
     batch_size_per_card: 128
     drop_last: true
-    num_workers: 1
+    num_workers: 4
 Eval:
   dataset:
     name: SimpleDataSet
@@ -126,4 +126,4 @@ Eval:
     shuffle: false
     drop_last: false
     batch_size_per_card: 128
-    num_workers: 1
+    num_workers: 4
diff --git a/Vision/OCR/korean_PP-OCRv3_rec.yml b/Vision/OCR/korean_PP-OCRv3_rec.yml
@@ -101,7 +101,7 @@ Train:
     shuffle: true
     batch_size_per_card: 128
     drop_last: true
-    num_workers: 1
+    num_workers: 4
 Eval:
   dataset:
     name: SimpleDataSet
@@ -126,4 +126,4 @@ Eval:
     shuffle: false
     drop_last: false
     batch_size_per_card: 128
-    num_workers: 1
+    num_workers: 4