feat: Updated Darknet-53 pretrained URL (#30)

* feat: Updated Darknet-53 pretrained model * docs: Updated classification reference README
frgfm · May 11, 2020 · d0c789d · d0c789d
1 parent da385c6
commit d0c789d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/holocron/models/darknet.py b/holocron/models/darknet.py
@@ -26,7 +26,7 @@
                   'url': 'https://github.com/frgfm/Holocron/releases/download/v0.1.0/darknet19_224-44aaede3.pth'},
     'darknet53': {'arch': 'DarknetV3',
                   'layout': [(1, 64, 128), (2, 128, 256), (8, 256, 512), (8, 512, 1024), (4, 1024)],
-                  'url': 'https://github.com/frgfm/Holocron/releases/download/v0.1.0/darknet53_224-f4c640a8.pth'},
+                  'url': 'https://github.com/frgfm/Holocron/releases/download/v0.1.0/darknet53_224-42576ca0.pth'},
 }
 
 

diff --git a/references/classification/README.md b/references/classification/README.md
@@ -35,14 +35,15 @@ python train.py imagenette2-320/ --model darknet53 --lr 5e-3 -b 32 -j 16 --epoch
 | 224       | 5      | imagenette2-320/ --model darknet53 --lr 5e-3 -b 32 -j 16 --epochs 5 --opt radam --sched onecycle --loss label_smoothing | 66.88%         | 1      |
 | 224       | 10     | imagenette2-320/ --model darknet53 --lr 5e-3 -b 32 -j 16 --epochs 10 --opt radam --sched onecycle --loss label_smoothing | 76.18%         | 1      |
 | 224       | 20     | imagenette2-320/ --model darknet19 --lr 5e-4 -b 32 -j 16 --epochs 20 --opt radam --sched onecycle --loss label_smoothing | 84.43%         | 1      |
+| 224       | 40     | imagenette2-320/ --model darknet53 --lr 5e-3 -b 32 -j 16 --epochs 40 --opt radam --sched onecycle --loss label_smoothing | 87.62%         | 1      |
 
 
 
 ## Model zoo
 
 | Model     | Accuracy@1 (Err) | Param # | MACs  | Interpolation | Image size |
 | --------- | ---------------- | ------- | ----- | ------------- | ---------- |
-| darknet53 | 82.57 (17.43)    | 40.60M  | 7.13G | bilinear      | 224        |
+| darknet53 | 87.62 (12.38)    | 40.60M  | 7.13G | bilinear      | 224        |
 | darknet19 | 84.43 (15.57)    | 19.83M  | 2.71G | bilinear      | 224        |
 | darnet24  | 78.75 (21.25)    | 22.40M  | 4.21G | bilinear      | 224        |