YOLO-World / third_party /mmyolo /configs /yolov5 /voc /yolov5_l-v61_fast_1xb32-50e_voc.py
stevengrove
initial commit
186701e
raw
history blame contribute delete
No virus
780 Bytes
_base_ = './yolov5_s-v61_fast_1xb64-50e_voc.py'
deepen_factor = 1.0
widen_factor = 1.0
train_batch_size_per_gpu = 32
train_num_workers = 8
load_from = 'https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_l-v61_syncbn_fast_8xb16-300e_coco/yolov5_l-v61_syncbn_fast_8xb16-300e_coco_20220917_031007-096ef0eb.pth' # noqa
model = dict(
backbone=dict(
deepen_factor=deepen_factor,
widen_factor=widen_factor,
),
neck=dict(
deepen_factor=deepen_factor,
widen_factor=widen_factor,
),
bbox_head=dict(head_module=dict(widen_factor=widen_factor)))
train_dataloader = dict(
batch_size=train_batch_size_per_gpu, num_workers=train_num_workers)
optim_wrapper = dict(
optimizer=dict(batch_size_per_gpu=train_batch_size_per_gpu))