fix mem oom

2021-11-10 16:09:09 +08:00 · 2021-11-10 16:09:09 +08:00 · 9cfdd331c4
parent 237febb143
commit 9cfdd331c4
1 changed files with 2 additions and 2 deletions
--- 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