mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
903b239f13
* add comet to logger interface * add comet logger * add support for updated parameters * clean up offline logger creation * update callback args for comet logger * add comet optimizer * add optimizer config * add comet README * update tutorial notebook with Comet section * add option to log class level metrics * add support for class level metrics and confusion matrix * handle errors when adding files to artifacts * fix typo * clean resume workflow * updates for HPO * update comet README * fix typo in comet README * update code snippets in comet README * update comet links in tutorial * updated links * change optimizer batch size param and update comet README image * update comet section in tutorial * use prexisting cmd line flags to configure logger * update artifact upload/download flow * remove come remove comet logger specific cmd line args * move downloading weights into comet logger code * remove extra argparse * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * change checkpoint logging flow to follow offline logger * update resume flow * add comet logger to remote dataset property * update cmd line args in hpo * set types for integer/float env variables * update README * fix typo in README * default to always logging model predictions * Update tutorial.ipynb * Update train.py * Add Comet to Integrations table * Update README.md * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
210 lines
2.9 KiB
JSON
210 lines
2.9 KiB
JSON
{
|
|
"algorithm": "random",
|
|
"parameters": {
|
|
"anchor_t": {
|
|
"type": "discrete",
|
|
"values": [
|
|
2,
|
|
8
|
|
]
|
|
},
|
|
"batch_size": {
|
|
"type": "discrete",
|
|
"values": [
|
|
16,
|
|
32,
|
|
64
|
|
]
|
|
},
|
|
"box": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.02,
|
|
0.2
|
|
]
|
|
},
|
|
"cls": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.2
|
|
]
|
|
},
|
|
"cls_pw": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.5
|
|
]
|
|
},
|
|
"copy_paste": {
|
|
"type": "discrete",
|
|
"values": [
|
|
1
|
|
]
|
|
},
|
|
"degrees": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0,
|
|
45
|
|
]
|
|
},
|
|
"epochs": {
|
|
"type": "discrete",
|
|
"values": [
|
|
5
|
|
]
|
|
},
|
|
"fl_gamma": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"fliplr": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"flipud": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"hsv_h": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"hsv_s": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"hsv_v": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"iou_t": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.7
|
|
]
|
|
},
|
|
"lr0": {
|
|
"type": "discrete",
|
|
"values": [
|
|
1e-05,
|
|
0.1
|
|
]
|
|
},
|
|
"lrf": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.01,
|
|
1
|
|
]
|
|
},
|
|
"mixup": {
|
|
"type": "discrete",
|
|
"values": [
|
|
1
|
|
]
|
|
},
|
|
"momentum": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.6
|
|
]
|
|
},
|
|
"mosaic": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"obj": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.2
|
|
]
|
|
},
|
|
"obj_pw": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0.5
|
|
]
|
|
},
|
|
"optimizer": {
|
|
"type": "categorical",
|
|
"values": [
|
|
"SGD",
|
|
"Adam",
|
|
"AdamW"
|
|
]
|
|
},
|
|
"perspective": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"scale": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"shear": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"translate": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0
|
|
]
|
|
},
|
|
"warmup_bias_lr": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0,
|
|
0.2
|
|
]
|
|
},
|
|
"warmup_epochs": {
|
|
"type": "discrete",
|
|
"values": [
|
|
5
|
|
]
|
|
},
|
|
"warmup_momentum": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0,
|
|
0.95
|
|
]
|
|
},
|
|
"weight_decay": {
|
|
"type": "discrete",
|
|
"values": [
|
|
0,
|
|
0.001
|
|
]
|
|
}
|
|
},
|
|
"spec": {
|
|
"maxCombo": 0,
|
|
"metric": "metrics/mAP_0.5",
|
|
"objective": "maximize"
|
|
},
|
|
"trials": 1
|
|
}
|