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/timm
Ross Wightman 6441e9cc1b
Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.
5 years ago
..
data added MultiEpochsDataLoader 5 years ago
loss Update AugMix, JSD, etc comments and references 5 years ago
models Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args. 5 years ago
optim Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults 5 years ago
scheduler Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args. 5 years ago
__init__.py Add model registry and model listing fns, refactor model_factory/create_model fn 5 years ago
utils.py Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111 5 years ago
version.py Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too. 5 years ago