[Fix] Revise high workers_per_gpus (#1506)
parent
239049cf91
commit
62c3a7dda9
|
@ -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,
|
||||
)
|
||||
|
|
|
@ -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,
|
||||
)
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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))
|
||||
|
|
Loading…
Reference in New Issue