You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/models
Ross Wightman 996c77aa94
Prep mnasnet for pretrained models, use the select global pool, some comment mistakes
5 years ago
..
__init__.py Uniform pretrained model handling. 5 years ago
adaptive_avgmax_pool.py Lots of refactoring and cleanup. 5 years ago
densenet.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
dpn.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
helpers.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
inception_resnet_v2.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
inception_v4.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
median_pool.py Initial commit, puting some ol pieces together 5 years ago
mnasnet.py Prep mnasnet for pretrained models, use the select global pool, some comment mistakes 5 years ago
model_factory.py Add MNASNet A1, B1, and Small models as per the TF impl. Testing/training in progress... 5 years ago
pnasnet.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
resnet.py Add checkpoint clean script, add link to pretrained resnext50 weights 5 years ago
senet.py Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn 5 years ago
test_time_pool.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago
xception.py Add per model crop pct, interpolation defaults, tie it all together 5 years ago