Skip to content

GitLab

Commit 49a1e555 authored by VVsssssk's avatar VVsssssk Committed by ChaimZhu
Browse files

fix batch size

parent 2155ff69
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
configs/_base_/datasets/kitti-3d-3class.py
View file @ 49a1e555
...@@ -87,8 +87,8 @@ eval_pipeline = [ ...@@ -87,8 +87,8 @@ eval_pipeline = [
dict(type='Pack3DDetInputs', keys=['points']), dict(type='Pack3DDetInputs', keys=['points']),
] ]
train_dataloader = dict( train_dataloader = dict(
batch_size=2, batch_size=6,
num_workers=2, num_workers=4,
persistent_workers=True, persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True), sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict( dataset=dict(
......
configs/_base_/datasets/kitti-3d-car.py
View file @ 49a1e555
...@@ -85,8 +85,8 @@ eval_pipeline = [ ...@@ -85,8 +85,8 @@ eval_pipeline = [
dict(type='Pack3DDetInputs', keys=['points']), dict(type='Pack3DDetInputs', keys=['points']),
] ]
train_dataloader = dict( train_dataloader = dict(
batch_size=2, batch_size=6,
num_workers=2, num_workers=4,
persistent_workers=True, persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True), sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict( dataset=dict(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment