|
|
@ -27,7 +27,7 @@ deterministic: True # (bool) whether to enable deterministic mode
|
|
|
|
single_cls: False # (bool) train multi-class data as single-class
|
|
|
|
single_cls: False # (bool) train multi-class data as single-class
|
|
|
|
rect: False # (bool) rectangular training if mode='train' or rectangular validation if mode='val'
|
|
|
|
rect: False # (bool) rectangular training if mode='train' or rectangular validation if mode='val'
|
|
|
|
cos_lr: False # (bool) use cosine learning rate scheduler
|
|
|
|
cos_lr: False # (bool) use cosine learning rate scheduler
|
|
|
|
close_mosaic: 0 # (int) disable mosaic augmentation for final epochs
|
|
|
|
close_mosaic: 10 # (int) disable mosaic augmentation for final epochs
|
|
|
|
resume: False # (bool) resume training from last checkpoint
|
|
|
|
resume: False # (bool) resume training from last checkpoint
|
|
|
|
amp: True # (bool) Automatic Mixed Precision (AMP) training, choices=[True, False], True runs AMP check
|
|
|
|
amp: True # (bool) Automatic Mixed Precision (AMP) training, choices=[True, False], True runs AMP check
|
|
|
|
fraction: 1.0 # (float) dataset fraction to train on (default is 1.0, all images in train set)
|
|
|
|
fraction: 1.0 # (float) dataset fraction to train on (default is 1.0, all images in train set)
|
|
|
|