bmombie commited on
Commit
210014b
1 Parent(s): f3fb667

Delete config.json

Browse files
Files changed (1) hide show
  1. config.json +0 -39
config.json DELETED
@@ -1,39 +0,0 @@
1
- {
2
- "architectures": [
3
- "YolosForObjectDetection"
4
- ],
5
- "model_type": "yolov8",
6
- "model_architecture": "yolov8n",
7
- "num_classes": 1,
8
- "id2label": {
9
- "0": "bicycle_rider"
10
- },
11
- "label2id": {
12
- "bicycle_rider": 0
13
- },
14
- "image_size": [
15
- 640,
16
- 640
17
- ],
18
- "num_channels": 3,
19
- "task": "detect",
20
- "mode": "train",
21
- "epochs": 25,
22
- "batch_size": 16,
23
- "optimizer": "auto",
24
- "learning_rate": 0.01,
25
- "lr_scheduler": {
26
- "type": "linear",
27
- "start_factor": 0.01,
28
- "end_factor": 1.0,
29
- "total_iters": 3
30
- },
31
- "weight_decay": 0.0005,
32
- "momentum": 0.937,
33
- "nms_threshold": 0.7,
34
- "confidence_threshold": 0.2,
35
- "device": "cuda",
36
- "is_pretrained": true,
37
- "input_size": 640,
38
- "transformers_version": "4.19.0.dev0"
39
- }