FredZhang7 commited on
Commit
a9e2ffc
·
1 Parent(s): c0931f3

add comparisons, fix typos

Browse files
Files changed (1) hide show
  1. README.md +8 -2
README.md CHANGED
@@ -6,7 +6,8 @@ tags:
6
  license: cc-by-nc-4.0
7
  ---
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
@@ -55,4 +56,9 @@ traced_model.save(model_name)
55
 
56
  # Load the training-ready model
57
  model = torch.load(model_name)
58
- ```
 
 
 
 
 
 
6
  license: cc-by-nc-4.0
7
  ---
8
 
9
+ To be clear, this model is tailored to my image and video classification tasks, not to imagenet.
10
+ I built EfficientNetV2.5 to outperform the existing EfficientNet b0 to b4 and EfficientNetV2 t to l models, whether in TensorFlow or PyTorch, in terms of top-1 accuracy, efficiency, and robustness on my datasets and GVNS benchmarks.
11
 
12
  ## Model Details
13
  - **Model tasks:** Image classification / video classification / feature backbone
 
56
 
57
  # Load the training-ready model
58
  model = torch.load(model_name)
59
+ ```
60
+
61
+ # Top-1 Accuracy Comparisons
62
+ I will publish the results in another model repository, including the link to the GVNS benchmark.
63
+
64
+ Note that `efficientnet_b3_pruned` achieved the second highest top-1 accuracy as well as the highest epoch-1 training accuracy on my task, out of all previous EfficientNet models my 24 GB VRAM RTX 3090 could handle.