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
88129b2569
Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
|
4 years ago |
| .. |
|
data
|
added MultiEpochsDataLoader
|
4 years ago |
|
loss
|
Update AugMix, JSD, etc comments and references
|
5 years ago |
|
models
|
Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
|
4 years ago |
|
optim
|
Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults
|
4 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.
|
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
|
4 years ago |
|
version.py
|
Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.
|
4 years ago |
