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 9eba134d79
More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.
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 More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 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