Deeplite · soudlai · Jun 10, 2024 · Jun 11, 2024 · Jun 11, 2024 · Jun 14, 2024
diff --git a/.travis.yml b/.travis.yml
@@ -32,10 +32,10 @@ script:
   - pylint deeplite_torch_zoo/utils/ --ignore-docstrings yes --ignore-comments yes --disable=W0614,W0401,W0613,C0301,E0401,C0415
   - pytest
   - codecov
-  - python training_scripts/object_detection/train.py --model_name yolo8n --dataset coco8 --dryrun
-  - python training_scripts/classification/ultralytics/train.py --model resnet12 --dataset imagewoof_160 --dryrun
-  - python training_scripts/classification/cifar/train_cifar.py --model resnet18 --dataset cifar100 --dryrun
-  - python training_scripts/classification/imagenet/train.py --data-dir ./ --dataset imagewoof_160 --model resnet12 --dryrun --dataset-download
+  - python deeplite_torch_zoo/training_scripts/object_detection/train.py --model yolo8n --dataset coco8 --dryrun
+  - python deeplite_torch_zoo/training_scripts/classification/ultralytics/train.py --model resnet12 --dataset imagewoof_160 --dryrun
+  - python deeplite_torch_zoo/training_scripts/classification/cifar/train_cifar.py --model resnet18 --dataset cifar100 --dryrun
+  - python deeplite_torch_zoo/training_scripts/classification/imagenet/train.py --data-dir ./ --dataset imagewoof_160 --model resnet12 --dryrun --dataset-download
 
 before_deploy:
   - pip install markupsafe==2.0.1

diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,2 +1,4 @@
 recursive-include deeplite_torch_zoo/src *.yaml
-recursive-include deeplite_torch_zoo/api *.yaml
+recursive-include deeplite_torch_zoo/api *.yaml
+recursive-include deeplite_torch_zoo/training_scripts *.yaml
+recursive-include deeplite_torch_zoo/training_scripts *.py
diff --git a/...g_scripts/classification/cifar/LICENSE.md → ...g_scripts/classification/cifar/LICENSE.md b/...g_scripts/classification/cifar/LICENSE.md → ...g_scripts/classification/cifar/LICENSE.md
diff --git a/...deeplite.localresnet50/events.out.tfevents.1718384432.deeplite05.deeplite.local.1625076.0 b/...deeplite.localresnet50/events.out.tfevents.1718384432.deeplite05.deeplite.local.1625076.0
diff --git a/...deeplite.localresnet50/events.out.tfevents.1718384445.deeplite05.deeplite.local.1625243.0 b/...deeplite.localresnet50/events.out.tfevents.1718384445.deeplite05.deeplite.local.1625243.0
diff --git a/...deeplite.localresnet50/events.out.tfevents.1718384491.deeplite05.deeplite.local.1625550.0 b/...deeplite.localresnet50/events.out.tfevents.1718384491.deeplite05.deeplite.local.1625550.0
diff --git a/...ripts/classification/cifar/train_cifar.py → ...ripts/classification/cifar/train_cifar.py b/...ripts/classification/cifar/train_cifar.py → ...ripts/classification/cifar/train_cifar.py
diff --git a/...g_scripts/classification/imagenet/LICENSE → ...g_scripts/classification/imagenet/LICENSE b/...g_scripts/classification/imagenet/LICENSE → ...g_scripts/classification/imagenet/LICENSE
diff --git a/...fication/imagenet/configs/ghostnetv2.yaml → ...fication/imagenet/configs/ghostnetv2.yaml b/...fication/imagenet/configs/ghostnetv2.yaml → ...fication/imagenet/configs/ghostnetv2.yaml
diff --git a/...ification/imagenet/configs/ra3-other.yaml → ...ification/imagenet/configs/ra3-other.yaml b/...ification/imagenet/configs/ra3-other.yaml → ...ification/imagenet/configs/ra3-other.yaml
diff --git a/.../classification/imagenet/configs/ra3.yaml → .../classification/imagenet/configs/ra3.yaml b/.../classification/imagenet/configs/ra3.yaml → .../classification/imagenet/configs/ra3.yaml
diff --git a/...ssification/imagenet/configs/timm_a1.yaml → ...ssification/imagenet/configs/timm_a1.yaml b/...ssification/imagenet/configs/timm_a1.yaml → ...ssification/imagenet/configs/timm_a1.yaml
diff --git a/...ssification/imagenet/configs/timm_a2.yaml → ...ssification/imagenet/configs/timm_a2.yaml b/...ssification/imagenet/configs/timm_a2.yaml → ...ssification/imagenet/configs/timm_a2.yaml
diff --git a/...ssification/imagenet/configs/timm_a3.yaml → ...ssification/imagenet/configs/timm_a3.yaml b/...ssification/imagenet/configs/timm_a3.yaml → ...ssification/imagenet/configs/timm_a3.yaml
diff --git a/...tion/imagenet/configs/timm_a3_img176.yaml → ...tion/imagenet/configs/timm_a3_img176.yaml b/...tion/imagenet/configs/timm_a3_img176.yaml → ...tion/imagenet/configs/timm_a3_img176.yaml
diff --git a/.../classification/imagenet/configs/usi.yaml → .../classification/imagenet/configs/usi.yaml b/.../classification/imagenet/configs/usi.yaml → .../classification/imagenet/configs/usi.yaml
diff --git a/...ssification/imagenet/distributed_train.sh → ...ssification/imagenet/distributed_train.sh b/...ssification/imagenet/distributed_train.sh → ...ssification/imagenet/distributed_train.sh
diff --git a/..._scripts/classification/imagenet/train.py → ..._scripts/classification/imagenet/train.py b/..._scripts/classification/imagenet/train.py → ..._scripts/classification/imagenet/train.py
diff --git a/...cripts/classification/ultralytics/LICENSE → ...cripts/classification/ultralytics/LICENSE b/...cripts/classification/ultralytics/LICENSE → ...cripts/classification/ultralytics/LICENSE
diff --git a/...ics/runs/train-cls/exp/events.out.tfevents.1718384616.deeplite05.deeplite.local.1626998.0 b/...ics/runs/train-cls/exp/events.out.tfevents.1718384616.deeplite05.deeplite.local.1626998.0
diff --git a/deeplite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp/opt.yaml b/deeplite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp/opt.yaml
@@ -0,0 +1,28 @@
+data_root: /neutrino/datasets/flower-102/
+model: resnet18
+dataset: flower102
+pretraining_dataset: flower102
+epochs: 1
+batch_size: 16
+test_batch_size: 256
+imgsz: 640
+nosave: false
+device: '[0,1,2,3]'
+workers: 8
+project: /home/sougata.pal/deeplite-torch-zoo/training_scripts/classification/ultralytics/runs/train-cls
+name: exp
+exist_ok: false
+pretrained: true
+optimizer: Adam
+lr0: 0.001
+decay: 5.0e-05
+label_smoothing: 0.1
+verbose: false
+seed: 0
+local_rank: -1
+kd_model_name: null
+kd_model_checkpoint: null
+alpha_kd: 5
+use_kd_loss_only: false
+dryrun: false
+save_dir: /home/sougata.pal/deeplite-torch-zoo/training_scripts/classification/ultralytics/runs/train-cls/exp
diff --git a/...cs/runs/train-cls/exp2/events.out.tfevents.1718384664.deeplite05.deeplite.local.1627420.0 b/...cs/runs/train-cls/exp2/events.out.tfevents.1718384664.deeplite05.deeplite.local.1627420.0
diff --git a/deeplite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/opt.yaml b/deeplite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/opt.yaml
@@ -0,0 +1,28 @@
+data_root: /neutrino/datasets/flowers-102/
+model: resnet18
+dataset: flowers102
+pretraining_dataset: flowers102
+epochs: 1
+batch_size: 16
+test_batch_size: 256
+imgsz: 640
+nosave: false
+device: '[0,1,2,3]'
+workers: 8
+project: /home/sougata.pal/deeplite-torch-zoo/training_scripts/classification/ultralytics/runs/train-cls
+name: exp
+exist_ok: false
+pretrained: true
+optimizer: Adam
+lr0: 0.001
+decay: 5.0e-05
+label_smoothing: 0.1
+verbose: false
+seed: 0
+local_rank: -1
+kd_model_name: null
+kd_model_checkpoint: null
+alpha_kd: 5
+use_kd_loss_only: false
+dryrun: false
+save_dir: /home/sougata.pal/deeplite-torch-zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2
diff --git a/...ite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/results.csv b/...ite_torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/results.csv
@@ -0,0 +1,2 @@
+                  epoch,             train/loss,  metrics/accuracy_top1,  metrics/accuracy_top5,                   lr/0
+                      0,                 2.4841,                0.55993,                0.81656,                  1e-05
diff --git a/...torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/weights/best.pt b/...torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/weights/best.pt
diff --git a/...raining_scripts/classification/ultralytics/runs/train-cls/exp2/weights/best_state_dict.pt b/...raining_scripts/classification/ultralytics/runs/train-cls/exp2/weights/best_state_dict.pt
diff --git a/...torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/weights/last.pt b/...torch_zoo/training_scripts/classification/ultralytics/runs/train-cls/exp2/weights/last.pt
diff --git a/...ripts/classification/ultralytics/train.py → ...ripts/classification/ultralytics/train.py b/...ripts/classification/ultralytics/train.py → ...ripts/classification/ultralytics/train.py
diff --git a/training_scripts/object_detection/LICENSE → ...training_scripts/object_detection/LICENSE b/training_scripts/object_detection/LICENSE → ...training_scripts/object_detection/LICENSE
diff --git a/training_scripts/object_detection/export.py → ...aining_scripts/object_detection/export.py b/training_scripts/object_detection/export.py → ...aining_scripts/object_detection/export.py
diff --git a/...object_detection/hyps/hyp.Objects365.yaml → ...object_detection/hyps/hyp.Objects365.yaml b/...object_detection/hyps/hyp.Objects365.yaml → ...object_detection/hyps/hyp.Objects365.yaml
diff --git a/...cripts/object_detection/hyps/hyp.VOC.yaml → ...cripts/object_detection/hyps/hyp.VOC.yaml b/...cripts/object_detection/hyps/hyp.VOC.yaml → ...cripts/object_detection/hyps/hyp.VOC.yaml
diff --git a/...t_detection/hyps/hyp.no-augmentation.yaml → ...t_detection/hyps/hyp.no-augmentation.yaml b/...t_detection/hyps/hyp.no-augmentation.yaml → ...t_detection/hyps/hyp.no-augmentation.yaml
diff --git a/...ject_detection/hyps/hyp.scratch-high.yaml → ...ject_detection/hyps/hyp.scratch-high.yaml b/...ject_detection/hyps/hyp.scratch-high.yaml → ...ject_detection/hyps/hyp.scratch-high.yaml
diff --git a/...bject_detection/hyps/hyp.scratch-low.yaml → ...bject_detection/hyps/hyp.scratch-low.yaml b/...bject_detection/hyps/hyp.scratch-low.yaml → ...bject_detection/hyps/hyp.scratch-low.yaml
diff --git a/...bject_detection/hyps/hyp.scratch-med.yaml → ...bject_detection/hyps/hyp.scratch-med.yaml b/...bject_detection/hyps/hyp.scratch-med.yaml → ...bject_detection/hyps/hyp.scratch-med.yaml
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp/hyp.yaml b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp/hyp.yaml
@@ -0,0 +1,28 @@
+lr0: 0.01
+lrf: 0.01
+momentum: 0.937
+weight_decay: 0.0005
+warmup_epochs: 3.0
+warmup_momentum: 0.8
+warmup_bias_lr: 0.1
+box: 0.05
+cls: 0.5
+cls_pw: 1.0
+obj: 1.0
+obj_pw: 1.0
+iou_t: 0.2
+anchor_t: 4.0
+fl_gamma: 0.0
+hsv_h: 0.015
+hsv_s: 0.7
+hsv_v: 0.4
+degrees: 0.0
+translate: 0.1
+scale: 0.5
+shear: 0.0
+perspective: 0.0
+flipud: 0.0
+fliplr: 0.5
+mosaic: 1.0
+mixup: 0.0
+copy_paste: 0.0
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp/opt.yaml b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp/opt.yaml
@@ -0,0 +1,64 @@
+model_name: yolo5n
+pretrained: true
+pretraining_dataset: coco
+data_root: /neutrino/datasets/VOCdevkit/
+dataset: voc
+hyp:
+  lr0: 0.01
+  lrf: 0.01
+  momentum: 0.937
+  weight_decay: 0.0005
+  warmup_epochs: 3.0
+  warmup_momentum: 0.8
+  warmup_bias_lr: 0.1
+  box: 0.05
+  cls: 0.5
+  cls_pw: 1.0
+  obj: 1.0
+  obj_pw: 1.0
+  iou_t: 0.2
+  anchor_t: 4.0
+  fl_gamma: 0.0
+  hsv_h: 0.015
+  hsv_s: 0.7
+  hsv_v: 0.4
+  degrees: 0.0
+  translate: 0.1
+  scale: 0.5
+  shear: 0.0
+  perspective: 0.0
+  flipud: 0.0
+  fliplr: 0.5
+  mosaic: 1.0
+  mixup: 0.0
+  copy_paste: 0.0
+epochs: 1
+batch_size: 16
+imgsz: 640
+rect: false
+nosave: false
+noval: false
+noautoanchor: false
+cache: null
+image_weights: false
+device: '[0,1,2,3]'
+multi_scale: false
+single_cls: false
+optimizer: SGD
+sync_bn: false
+workers: 8
+project: runs/train
+name: exp
+exist_ok: false
+quad: false
+cos_lr: false
+label_smoothing: 0.0
+patience: 100
+freeze:
+- 0
+save_period: -1
+seed: 0
+no_amp: false
+local_rank: -1
+dryrun: false
+save_dir: runs/train/exp
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/hyp.yaml b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/hyp.yaml
@@ -0,0 +1,28 @@
+lr0: 0.01
+lrf: 0.01
+momentum: 0.937
+weight_decay: 0.0005
+warmup_epochs: 3.0
+warmup_momentum: 0.8
+warmup_bias_lr: 0.1
+box: 0.05
+cls: 0.5
+cls_pw: 1.0
+obj: 1.0
+obj_pw: 1.0
+iou_t: 0.2
+anchor_t: 4.0
+fl_gamma: 0.0
+hsv_h: 0.015
+hsv_s: 0.7
+hsv_v: 0.4
+degrees: 0.0
+translate: 0.1
+scale: 0.5
+shear: 0.0
+perspective: 0.0
+flipud: 0.0
+fliplr: 0.5
+mosaic: 1.0
+mixup: 0.0
+copy_paste: 0.0
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/opt.yaml b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/opt.yaml
@@ -0,0 +1,64 @@
+model_name: yolo5n
+pretrained: true
+pretraining_dataset: voc
+data_root: /neutrino/datasets/VOCdevkit/
+dataset: voc
+hyp:
+  lr0: 0.01
+  lrf: 0.01
+  momentum: 0.937
+  weight_decay: 0.0005
+  warmup_epochs: 3.0
+  warmup_momentum: 0.8
+  warmup_bias_lr: 0.1
+  box: 0.05
+  cls: 0.5
+  cls_pw: 1.0
+  obj: 1.0
+  obj_pw: 1.0
+  iou_t: 0.2
+  anchor_t: 4.0
+  fl_gamma: 0.0
+  hsv_h: 0.015
+  hsv_s: 0.7
+  hsv_v: 0.4
+  degrees: 0.0
+  translate: 0.1
+  scale: 0.5
+  shear: 0.0
+  perspective: 0.0
+  flipud: 0.0
+  fliplr: 0.5
+  mosaic: 1.0
+  mixup: 0.0
+  copy_paste: 0.0
+epochs: 1
+batch_size: 16
+imgsz: 640
+rect: false
+nosave: false
+noval: false
+noautoanchor: false
+cache: null
+image_weights: false
+device: '[0,1,2,3]'
+multi_scale: false
+single_cls: false
+optimizer: SGD
+sync_bn: false
+workers: 8
+project: runs/train
+name: exp
+exist_ok: false
+quad: false
+cos_lr: false
+label_smoothing: 0.0
+patience: 100
+freeze:
+- 0
+save_period: -1
+seed: 0
+no_amp: false
+local_rank: -1
+dryrun: false
+save_dir: runs/train/exp2
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/weights/best.pt b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/weights/best.pt
diff --git a/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/weights/last.pt b/deeplite_torch_zoo/training_scripts/object_detection/runs/train/exp2/weights/last.pt
diff --git a/training_scripts/object_detection/train.py → ...raining_scripts/object_detection/train.py b/training_scripts/object_detection/train.py → ...raining_scripts/object_detection/train.py
@@ -64,7 +64,7 @@ def train(opt, device):  # hyp is path/to/hyp.yaml or hyp dictionary
 
     # Image size
     model = get_model(
-        model_name=opt.model_name,
+        model_name=opt.model,
         dataset_name=opt.pretraining_dataset,
         pretrained=False,
     ).to(device)
@@ -105,7 +105,7 @@ def train(opt, device):  # hyp is path/to/hyp.yaml or hyp dictionary
 
     # Model
     model = get_model(
-        model_name=opt.model_name,
+        model_name=opt.model,
         dataset_name=opt.pretraining_dataset,
         pretrained=opt.pretrained,
         num_classes=nc,
@@ -163,7 +163,7 @@ def train(opt, device):  # hyp is path/to/hyp.yaml or hyp dictionary
 
     eval_function = get_eval_function(
         dataset_name=opt.dataset,
-        model_name=opt.model_name,
+        model_name=opt.model,
     )
 
     # DDP mode
@@ -352,7 +352,7 @@ def train(opt, device):  # hyp is path/to/hyp.yaml or hyp dictionary
 
 def parse_opt(known=False):
     parser = argparse.ArgumentParser()
-    parser.add_argument('--model_name', type=str, default='yolo5s', help='zoo model name')
+    parser.add_argument('--model', type=str, default='yolo5s', help='zoo model name')
     parser.add_argument('--pretrained', action='store_true')
     parser.add_argument('--pretraining_dataset', type=str, default='coco')
     parser.add_argument('--data_root', type=str, default=None, help='dataset path')

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
     license = f.read()
 
 with open('README.md') as f:
-    long_description = f.read()
+    long_description = "" # f.read()
 
 
 # Package
@@ -33,7 +33,7 @@
     'pandas',
     'tensorboard>=2.11.2',
     'openmim==0.3.7',
-    'super-gradients==3.3.0',
+    'super-gradients==3.3.0'
 ]
 
 if sys.version_info.minor > 7: