FredZhang7
commited on
Commit
•
642ba04
1
Parent(s):
17b3b4c
add carbon emissions
Browse files
README.md
CHANGED
@@ -59,10 +59,13 @@ traced_model.save(model_name)
|
|
59 |
model = torch.load(model_name)
|
60 |
```
|
61 |
|
62 |
-
|
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 all previous 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.
|
|
|
|
|
|
|
|
59 |
model = torch.load(model_name)
|
60 |
```
|
61 |
|
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 all previous 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.
|
69 |
+
|
70 |
+
### Carbon Emissions
|
71 |
+
Comparing all models and testing my new architectures costed roughly 504 GPU hours, over a span of 27 days.
|