alihan/open_clip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Adding ResNet-101 checkpoint

Browse Source
This commit is contained in:
Gabriel Ilharco Magalhães
2021-08-03 22:28:52 -07:00
committed by GitHub
parent ed16e38a6b
commit a02e6d9385
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@@ -111,7 +111,10 @@ python src/training/main.py \
## Trained models
You can find our ResNet-50 trained on YFCC-15M [here](https://drive.google.com/file/d/1bbNMrWAq3NxCAteHmbrYgBKpGAA9j9pn/view?usp=sharing).
Below are checkpoints of models trained on YFCC-15M, along with their zero-shot top-1 accuracies on ImageNet and ImageNetV2. These models were trained using 8 GPUs and the same hyperparameters described in the "Sample running code" section, with the exception of `lr=5e-4` and `epochs=32`.
* [ResNet-50](https://drive.google.com/file/d/1bbNMrWAq3NxCAteHmbrYgBKpGAA9j9pn/view?usp=sharing) (32.7% / 27.9%)
* [ResNet-101](https://drive.google.com/file/d/1vOorR3pKkuuA_Gv7OgqMtaETNjTmy_3m/view?usp=sharing) (34.8% / 30.0%)
## Scaling trends