thrash_can_instance_segmentation/dvc.lock at main · FirstGalacticEmpire/thrash_can_instance_segmentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
schema: '2.0'
stages:
  train_yolact:
    cmd: python src/models/yolact/custom_train.py --config=yolact_resnet50_config
      --batch_size=3
    deps:
    - path: data/TrashCan/instances_train_trashcan.json
      md5: 9961b3b9982d913319322ae85b1f4ae6
      size: 22598259
    - path: data/TrashCan/instances_val_trashcan_splt.json
      md5: d0246f701303b104cb31ffbf2e09c2d6
      size: 3937280
    - path: data/TrashCan/train/
      md5: a75f86cd5816b041b6343f9778f8239a.dir
      size: 146143311
      nfiles: 6065
    - path: data/TrashCan/val/
      md5: 470480a35554fe1fc9f03f71c307f02d.dir
      size: 27222494
      nfiles: 1147
    - path: src/models/yolact/custom_train.py
      md5: a7c1923142e0e203b3930e43509dc598
      size: 16826
    - path: src/models/yolact/data/config.py
      md5: 59dc0d66771506b249770ee9b28a948d
      size: 33551
    outs:
    - path: models/yolact/best_model.pth
      md5: 507f5d08f7ff43eaa0223af91c06061a
      size: 123330173
  train_yolact53:
    cmd: python src/models/yolact/custom_train.py --config=yolact_darknet53_config
      --batch_size=2 --best_model=best_model53.pth --max_iter=20000
    deps:
    - path: data/TrashCan/instances_train_trashcan.json
      md5: 9961b3b9982d913319322ae85b1f4ae6
      size: 22598259
    - path: data/TrashCan/instances_val_trashcan_splt.json
      md5: d0246f701303b104cb31ffbf2e09c2d6
      size: 3937280
    - path: src/models/yolact/custom_train.py
      md5: a7c1923142e0e203b3930e43509dc598
      size: 16826
    - path: src/models/yolact/data/config.py
      md5: 59dc0d66771506b249770ee9b28a948d
      size: 33551
    outs:
    - path: models/yolact/best_model53.pth
      md5: a19003ecf353b0c27d4ac509b8757cd7
      size: 189736457
  overfit_yolact:
    cmd: python src/models/yolact/custom_train.py --config=yolact_resnet50_config_overfit
      --batch_size=3 --best_model=best_model_overfit.pth
    deps:
    - path: data/TrashCan/instances_overfit_trashcan.json
      md5: ba342879e77dcf85bd815387a61078c6
      size: 409567
    - path: data/TrashCan/train/
      md5: a75f86cd5816b041b6343f9778f8239a.dir
      size: 146143311
      nfiles: 6065
    - path: data/TrashCan/val/
      md5: 470480a35554fe1fc9f03f71c307f02d.dir
      size: 27222494
      nfiles: 1147
    - path: src/models/yolact/custom_train.py
      md5: a7c1923142e0e203b3930e43509dc598
      size: 16826
    - path: src/models/yolact/data/config.py
      md5: 59dc0d66771506b249770ee9b28a948d
      size: 33551
    outs:
    - path: models/yolact/best_model_overfit.pth
      md5: ab47e9ee4dd608235f6c463b619a08a4
      size: 123330173
  train_yolact_adam:
    cmd: python src/models/yolact/custom_train.py --config=yolact_resnet50_config
      --optimizer=adam --batch_size=3 --best_model=best_model_adam.pth
    deps:
    - path: data/TrashCan/instances_train_trashcan.json
      md5: 9961b3b9982d913319322ae85b1f4ae6
      size: 22598259
    - path: data/TrashCan/instances_val_trashcan_splt.json
      md5: d0246f701303b104cb31ffbf2e09c2d6
      size: 3937280
    - path: src/models/yolact/custom_train.py
      md5: a7c1923142e0e203b3930e43509dc598
      size: 16826
    - path: src/models/yolact/data/config.py
      md5: 59dc0d66771506b249770ee9b28a948d
      size: 33551
    outs:
    - path: models/yolact/best_model_adam.pth
      md5: 66df83aa854dcf63e043f398d9a7b4b0
      size: 123330173