Ross Wightman c099374771 Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42		2019-10-19 17:27:46 -07:00
..
__init__.py	Add Res2Net and DLA models w/ pretrained weights. Update sotabench.	2019-09-04 17:06:42 -07:00
adaptive_avgmax_pool.py	…
conv2d_helpers.py	Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl	2019-07-25 11:42:01 -07:00
densenet.py	Fix typo in Densenet default resolutions	2019-07-03 22:11:26 -07:00
dla.py	Add Res2Net and DLA models w/ pretrained weights. Update sotabench.	2019-09-04 17:06:42 -07:00
dpn.py	Register dpn107	2019-06-30 09:57:06 -07:00
factory.py	…
gen_efficientnet.py	Add memory efficient Swish impl	2019-10-19 14:48:30 -07:00
gluon_resnet.py	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.	2019-06-29 15:50:33 -07:00
gluon_xception.py	A few more additions to Gluon Xception models to match interface of others.	2019-08-11 23:06:23 -07:00
helpers.py	Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42	2019-10-19 17:27:46 -07:00
inception_resnet_v2.py	…
inception_v3.py	Update version, results csv files, and move remaining dropbox weights to github	2019-07-25 16:54:44 -07:00
inception_v4.py	…
median_pool.py	…
nasnet.py	Add NASNet-Large model	2019-07-03 22:10:50 -07:00
pnasnet.py	…
registry.py	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.	2019-06-29 15:50:33 -07:00
res2net.py	Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.	2019-09-05 15:47:56 -07:00
resnet.py	Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights.	2019-10-19 17:05:37 -07:00
senet.py	Update version, results csv files, and move remaining dropbox weights to github	2019-07-25 16:54:44 -07:00
test_time_pool.py	…
xception.py	…