| data: | |
| aug_param: | |
| flip_lr: 0.5 | |
| hsv: 0.3 | |
| mini_mosaic: 0.5 | |
| neg: 0.3 | |
| size_range: | |
| - 0.85 | |
| - 1.1 | |
| augment: true | |
| cache: false | |
| imgsz: 1024 | |
| train_img_dir: | |
| - ../datasets/codat_manga_v3/images/train | |
| - ../datasets/ComicErased/processed | |
| train_mask_dir: ../datasets/ComicSegV2 | |
| val_img_dir: | |
| - ../datasets/codat_manga_v3/images/val | |
| val_mask_dir: ../datasets/ComicSegV2 | |
| logger: | |
| project: '' | |
| run_id: '' | |
| type: wandb | |
| model: | |
| act: leaky | |
| weights: data/yolov5sblk.ckpt | |
| resume: | |
| ckpt: '' | |
| resume_training: false | |
| train: | |
| accumulation_steps: 4 | |
| batch_size: 4 | |
| epochs: 120 | |
| eval_interval: 1 | |
| linear_lr: false | |
| loss: dice | |
| lr0: 0.004 | |
| lrf: 0.005 | |
| momentum: 0.937 | |
| optimizer: adam | |
| warmup_bias_lr: 0.1 | |
| warmup_epochs: 3.0 | |
| warmup_momentum: 0.8 | |
| weight_decay: 2.0e-05 | |