FredZhang7
commited on
Commit
•
c034e97
1
Parent(s):
7b2bc87
improve clarity x 2
Browse files
README.md
CHANGED
@@ -62,7 +62,7 @@ model = torch.load(model_name)
|
|
62 |
### Top-1 Accuracy Comparisons
|
63 |
I finetuned the existing models on either 299x299, 304x304, 320x320, or 384x384 resolution, depending on the input size used during pretraining and the VRAM usage.
|
64 |
|
65 |
-
|
66 |
|
67 |
I will publish the detailed report in another model repository, including the link to the GVNS benchmarks.
|
68 |
This repository is only for the base model, pretrained on ImageNet, not my task.
|
|
|
62 |
### Top-1 Accuracy Comparisons
|
63 |
I finetuned the existing models on either 299x299, 304x304, 320x320, or 384x384 resolution, depending on the input size used during pretraining and the VRAM usage.
|
64 |
|
65 |
+
`efficientnet_b3_pruned` achieved the second highest top-1 accuracy as well as the highest epoch-1 training accuracy on my task, out of EfficientNetV2.5 small and all existing EfficientNet models my 24 GB VRAM RTX 3090 could handle.
|
66 |
|
67 |
I will publish the detailed report in another model repository, including the link to the GVNS benchmarks.
|
68 |
This repository is only for the base model, pretrained on ImageNet, not my task.
|