mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Revise high workers_per_gpus (#1506)

pull/1519/head
Miao Zheng 2022-04-25 13:19:47 +08:00 committed by GitHub
parent 239049cf91
commit 62c3a7dda9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py
View File

@ -1,5 +1,5 @@
_base_ = './bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py' _base_ = './bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py'
data = dict( data = dict(
samples_per_gpu=8, samples_per_gpu=8,
workers_per_gpu=8, workers_per_gpu=4,
) )

2
configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py
View File

@ -7,5 +7,5 @@ lr_config = dict(warmup='linear', warmup_iters=1000)
optimizer = dict(lr=0.05) optimizer = dict(lr=0.05)
data = dict( data = dict(
samples_per_gpu=8, samples_per_gpu=8,
workers_per_gpu=8, workers_per_gpu=4,
) )

2
configs/cgnet/cgnet_512x1024_60k_cityscapes.py
View File

@ -45,7 +45,7 @@ test_pipeline = [
] ]
data = dict( data = dict(
samples_per_gpu=8, samples_per_gpu=8,
workers_per_gpu=8, workers_per_gpu=4,
train=dict( train=dict(
type=dataset_type, type=dataset_type,
data_root=data_root, data_root=data_root,

2
configs/cgnet/cgnet_680x680_60k_cityscapes.py
View File

@ -44,7 +44,7 @@ test_pipeline = [
] ]
data = dict( data = dict(
samples_per_gpu=8, samples_per_gpu=8,
workers_per_gpu=8, workers_per_gpu=4,
train=dict(pipeline=train_pipeline), train=dict(pipeline=train_pipeline),
val=dict(pipeline=test_pipeline), val=dict(pipeline=test_pipeline),
test=dict(pipeline=test_pipeline)) test=dict(pipeline=test_pipeline))