FredZhang7 commited on
Commit
19fce4d
1 Parent(s): 7db28a3

add model details

Browse files
Files changed (1) hide show
  1. README.md +12 -0
README.md CHANGED
@@ -8,6 +8,18 @@ license: cc-by-nc-4.0
8
 
9
  To be clear, this model is tailored to my image and video classification tasks, not to imagenet. I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b7 and EfficientNetV2 t to xl models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and benchmarks.
10
 
 
 
 
 
 
 
 
 
 
 
 
 
11
 
12
  To change the number of classes, replace the linear classification layer.
13
  Here's an example to convert the architecture into a training-ready model.
 
8
 
9
  To be clear, this model is tailored to my image and video classification tasks, not to imagenet. I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b7 and EfficientNetV2 t to xl models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and benchmarks.
10
 
11
+ ## Model Details
12
+ - **Model tasks:** Image classification / video classification / feature backbone
13
+ - **Model stats:**
14
+ - Params: 16.64 M
15
+ - Multiply-Add Operations: 5.32 G
16
+ - Image size: train = 299x299 / 304x304, test = 304x304
17
+ - **Papers:**
18
+ - EfficientNetV2: Smaller Models and Faster Training: https://arxiv.org/abs/2104.00298
19
+ - Layer-adaptive sparsity for the Magnitude-based Pruning: https://arxiv.org/abs/2010.07611
20
+ - **Dataset:** ImageNet-1k
21
+ - **Pretrained:** Yes, but requires finetuning
22
+ - **Original:** This model architecture is original
23
 
24
  To change the number of classes, replace the linear classification layer.
25
  Here's an example to convert the architecture into a training-ready model.