From 62c3a7dda9aab13481760c9aa9e494001ea1fa89 Mon Sep 17 00:00:00 2001 From: Miao Zheng <76149310+MeowZheng@users.noreply.github.com> Date: Mon, 25 Apr 2022 13:19:47 +0800 Subject: [PATCH] [Fix] Revise high workers_per_gpus (#1506) --- .../bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py | 2 +- .../bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py | 2 +- configs/cgnet/cgnet_512x1024_60k_cityscapes.py | 2 +- configs/cgnet/cgnet_680x680_60k_cityscapes.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py b/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py index ea27ef0a1..f4b9f6d2c 100644 --- a/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py +++ b/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py @@ -1,5 +1,5 @@ _base_ = './bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py' data = dict( samples_per_gpu=8, - workers_per_gpu=8, + workers_per_gpu=4, ) diff --git a/configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py b/configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py index babc2cd74..5e93bea8b 100644 --- a/configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py +++ b/configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py @@ -7,5 +7,5 @@ lr_config = dict(warmup='linear', warmup_iters=1000) optimizer = dict(lr=0.05) data = dict( samples_per_gpu=8, - workers_per_gpu=8, + workers_per_gpu=4, ) diff --git a/configs/cgnet/cgnet_512x1024_60k_cityscapes.py b/configs/cgnet/cgnet_512x1024_60k_cityscapes.py index 11421ef9d..6efc4cd5c 100644 --- a/configs/cgnet/cgnet_512x1024_60k_cityscapes.py +++ b/configs/cgnet/cgnet_512x1024_60k_cityscapes.py @@ -45,7 +45,7 @@ test_pipeline = [ ] data = dict( samples_per_gpu=8, - workers_per_gpu=8, + workers_per_gpu=4, train=dict( type=dataset_type, data_root=data_root, diff --git a/configs/cgnet/cgnet_680x680_60k_cityscapes.py b/configs/cgnet/cgnet_680x680_60k_cityscapes.py index 2b2f8eefb..7d4ebe136 100644 --- a/configs/cgnet/cgnet_680x680_60k_cityscapes.py +++ b/configs/cgnet/cgnet_680x680_60k_cityscapes.py @@ -44,7 +44,7 @@ test_pipeline = [ ] data = dict( samples_per_gpu=8, - workers_per_gpu=8, + workers_per_gpu=4, train=dict(pipeline=train_pipeline), val=dict(pipeline=test_pipeline), test=dict(pipeline=test_pipeline))