fix mem oom

ga546 · Nov 10, 2021 · 9cfdd33 · 9cfdd33
1 parent 237febb
commit 9cfdd33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/det/det_r50_vd_sast_icdar15.yml b/configs/det/det_r50_vd_sast_icdar15.yml
@@ -93,7 +93,7 @@ Eval:
           channel_first: False
       - DetLabelEncode: # Class handling label
       - DetResizeForTest:
-          resize_long: 1536
+          resize_long: 768
       - NormalizeImage:
           scale: 1./255.
           mean: [0.485, 0.456, 0.406]
@@ -106,4 +106,4 @@ Eval:
     shuffle: False
     drop_last: False
     batch_size_per_card: 1 # must be 1
-    num_workers: 2
+    num_workers: 2