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
f122f0274b
Significant ResNet refactor:
...
* stage creation + make_layer moved to separate fn with more sensible dilation/output_stride calc
* drop path rate decay easy to impl with refactored block creation loops
* fix dilation + blur pool combo
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
Significant ResNet refactor:
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