[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

View File

@ -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,
)

View File

@ -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,
)

View File

@ -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,

View File

@ -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))