Upload args.yaml with huggingface_hub
Browse files
args.yaml
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task: detect
|
| 2 |
+
mode: train
|
| 3 |
+
model: outputs/pruning_direct/p30/pruned.pt
|
| 4 |
+
data: configs/data/deeppcb.yaml
|
| 5 |
+
epochs: 50
|
| 6 |
+
time: null
|
| 7 |
+
patience: 10
|
| 8 |
+
batch: 64
|
| 9 |
+
imgsz: 640
|
| 10 |
+
save: true
|
| 11 |
+
save_period: -1
|
| 12 |
+
cache: false
|
| 13 |
+
device: '0'
|
| 14 |
+
workers: 8
|
| 15 |
+
project: /kaggle/PCB-Prune-YOLO/outputs/finetune_direct
|
| 16 |
+
name: p30_adamw_exact
|
| 17 |
+
exist_ok: false
|
| 18 |
+
pretrained: true
|
| 19 |
+
cls_remap: true
|
| 20 |
+
optimizer: AdamW
|
| 21 |
+
verbose: true
|
| 22 |
+
seed: 42
|
| 23 |
+
deterministic: true
|
| 24 |
+
single_cls: false
|
| 25 |
+
rect: false
|
| 26 |
+
cos_lr: false
|
| 27 |
+
close_mosaic: 10
|
| 28 |
+
resume: false
|
| 29 |
+
amp: true
|
| 30 |
+
fraction: 1.0
|
| 31 |
+
profile: false
|
| 32 |
+
freeze: null
|
| 33 |
+
multi_scale: 0.0
|
| 34 |
+
compile: false
|
| 35 |
+
channels_last: false
|
| 36 |
+
overlap_mask: true
|
| 37 |
+
mask_ratio: 4
|
| 38 |
+
dropout: 0.0
|
| 39 |
+
val: true
|
| 40 |
+
split: val
|
| 41 |
+
save_json: false
|
| 42 |
+
conf: null
|
| 43 |
+
iou: 0.7
|
| 44 |
+
max_det: 300
|
| 45 |
+
quantize: null
|
| 46 |
+
dnn: false
|
| 47 |
+
plots: true
|
| 48 |
+
end2end: null
|
| 49 |
+
source: null
|
| 50 |
+
vid_stride: 1
|
| 51 |
+
stream_buffer: false
|
| 52 |
+
visualize: false
|
| 53 |
+
augment: false
|
| 54 |
+
agnostic_nms: false
|
| 55 |
+
classes: null
|
| 56 |
+
retina_masks: false
|
| 57 |
+
embed: null
|
| 58 |
+
show: false
|
| 59 |
+
save_frames: false
|
| 60 |
+
save_txt: false
|
| 61 |
+
save_conf: false
|
| 62 |
+
save_crop: false
|
| 63 |
+
show_labels: true
|
| 64 |
+
show_conf: true
|
| 65 |
+
show_boxes: true
|
| 66 |
+
line_width: null
|
| 67 |
+
format: torchscript
|
| 68 |
+
keras: false
|
| 69 |
+
optimize: false
|
| 70 |
+
dynamic: false
|
| 71 |
+
simplify: true
|
| 72 |
+
opset: null
|
| 73 |
+
workspace: null
|
| 74 |
+
nms: false
|
| 75 |
+
lr0: 0.001
|
| 76 |
+
lrf: 0.01
|
| 77 |
+
momentum: 0.9
|
| 78 |
+
weight_decay: 0.0005
|
| 79 |
+
warmup_epochs: 3.0
|
| 80 |
+
warmup_momentum: 0.8
|
| 81 |
+
warmup_bias_lr: 0.1
|
| 82 |
+
distill_model: null
|
| 83 |
+
dis: 6.0
|
| 84 |
+
box: 7.5
|
| 85 |
+
cls: 0.5
|
| 86 |
+
cls_pw: 0.0
|
| 87 |
+
dfl: 1.5
|
| 88 |
+
pose: 12.0
|
| 89 |
+
kobj: 1.0
|
| 90 |
+
rle: 1.0
|
| 91 |
+
angle: 1.0
|
| 92 |
+
dlog: 1.0
|
| 93 |
+
dgrad: 0.5
|
| 94 |
+
dlam: 1.0
|
| 95 |
+
nbs: 64
|
| 96 |
+
hsv_h: 0.015
|
| 97 |
+
hsv_s: 0.7
|
| 98 |
+
hsv_v: 0.4
|
| 99 |
+
degrees: 0.0
|
| 100 |
+
translate: 0.1
|
| 101 |
+
scale: 0.5
|
| 102 |
+
shear: 0.0
|
| 103 |
+
perspective: 0.0
|
| 104 |
+
flipud: 0.0
|
| 105 |
+
fliplr: 0.5
|
| 106 |
+
bgr: 0.0
|
| 107 |
+
mosaic: 1.0
|
| 108 |
+
mixup: 0.0
|
| 109 |
+
cutmix: 0.0
|
| 110 |
+
copy_paste: 0.0
|
| 111 |
+
copy_paste_mode: flip
|
| 112 |
+
auto_augment: randaugment
|
| 113 |
+
erasing: 0.4
|
| 114 |
+
cfg: null
|
| 115 |
+
tracker: tracktrack.yaml
|
| 116 |
+
save_dir: /kaggle/PCB-Prune-YOLO/outputs/finetune_direct/p30_adamw_exact
|