You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
384 B
YAML
24 lines
384 B
YAML
model: yolov5s-cls.pt
|
|
data: helmet_cls3
|
|
epochs: 80
|
|
batch_size: 504
|
|
imgsz: 224
|
|
nosave: false
|
|
cache: ram
|
|
device: ''
|
|
workers: 8
|
|
project: runs\train-cls
|
|
name: dduk_cls3
|
|
exist_ok: false
|
|
pretrained: true
|
|
optimizer: Adam
|
|
lr0: 0.001
|
|
decay: 5.0e-05
|
|
label_smoothing: 0.1
|
|
cutoff: null
|
|
dropout: null
|
|
verbose: false
|
|
seed: 0
|
|
local_rank: -1
|
|
save_dir: runs\train-cls\dduk_cls3
|