From 9cfdd331c4e9251fbe06617657d8da28337b0951 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Wed, 10 Nov 2021 16:09:09 +0800 Subject: [PATCH] fix mem oom --- configs/det/det_r50_vd_sast_icdar15.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/det/det_r50_vd_sast_icdar15.yml b/configs/det/det_r50_vd_sast_icdar15.yml index 674de7fb..a2255528 100755 --- 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 \ No newline at end of file + num_workers: 2 -- GitLab