Skip to content

Imagenet models weights

Compare
Choose a tag to compare
@bonlime bonlime released this 16 Apr 15:39

Releasing weights for models from
TResNet: High Performance GPU-Dedicated Architecture: https://arxiv.org/pdf/2003.13630.pdf
Original code: https://github.com/mrT23/TResNet
This weights were trained by authors and only ported to be loaded into models in this repo

Weights for HRNet were trained by authors as were uploaded to GitHub instead of OneDrive by me for easier loading.

Weights for Group Norm + Weight Standardization from original authors https://github.com/joe-siyuan-qiao/pytorch-classification/ . State dicts were remapped to allow easy loading into this repo models

Weights for Group Norm models R-50 & R-101 are from FAIR. They were originally trained in Caffe then ported to Detectron2 and then remapped by me to allow loading into this repo models.