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.

Ross Wightman fac58f609a Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak. ... * Add some of the trendy new optimizers. Decent results but not clearly better than the standards. * Can create a None scheduler for constant LR * ResNet defaults to zero_init of last BN in residual * add resnet50d config		5 years ago
..
__init__.py	Add working Gluon Xception-65 model. Some cleanup still needed.	5 years ago
adaptive_avgmax_pool.py	Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py	5 years ago
conv2d_helpers.py	Add MixNet (https://arxiv.org/abs/1907.09595) with pretrained weights converted from Tensorflow impl	5 years ago
densenet.py	Fix typo in Densenet default resolutions	5 years ago
dpn.py	Register dpn107	5 years ago
factory.py	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago
gen_efficientnet.py	Add weights for my MixNet-XL creation, include README updates for EdgeTPU models	5 years ago
gluon_resnet.py	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.	5 years ago
gluon_xception.py	A few more additions to Gluon Xception models to match interface of others.	5 years ago
helpers.py	Update helpers.py	5 years ago
inception_resnet_v2.py	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago
inception_v3.py	Update version, results csv files, and move remaining dropbox weights to github	5 years ago
inception_v4.py	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago
median_pool.py	Big re-org, working towards making pip/module as 'timm'	5 years ago
nasnet.py	Add NASNet-Large model	5 years ago
pnasnet.py	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago
registry.py	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.	5 years ago
resnet.py	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.	5 years ago
senet.py	Update version, results csv files, and move remaining dropbox weights to github	5 years ago
test_time_pool.py	Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py	5 years ago
xception.py	Add model registry and model listing fns, refactor model_factory/create_model fn	5 years ago