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 4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript		4 years ago
..
data	Working on feature extraction, interfaces refined, a number of models working, some in progress.	4 years ago
loss	Update AugMix, JSD, etc comments and references	5 years ago
models	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	4 years ago
optim	Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults	5 years ago
scheduler	Update README with model results and attribution. Make scheduler factory bit more robust to arg differences, add noise to plateau lr and fix min/max.	4 years ago
__init__.py	Monster commit, activation refactor, VoVNet, norm_act improvements, more	4 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	Remove tests from distrib	4 years ago