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
b093dcb46d
Some convnext cleanup, remove in place mul_ for gamma, breaking symbolic trace, cleanup head a bit...
|
3 years ago |
.. |
data
|
Fix MobileNetV2 head conv size for multiplier < 1.0. Add some missing modification copyrights, fix starting date of some old ones.
|
3 years ago |
loss
|
Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1)
|
3 years ago |
models
|
Some convnext cleanup, remove in place mul_ for gamma, breaking symbolic trace, cleanup head a bit...
|
3 years ago |
optim
|
Add lamb/lars to optim init imports, remove stray comment
|
3 years ago |
scheduler
|
Fix k_decay default arg != 1.0 in poly scheduler
|
3 years ago |
utils
|
Fix some formatting in utils/model.py
|
3 years ago |
__init__.py
|
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
|
3 years ago |
version.py
|
FBNetV3 timm trained weights added for b/d/g variants. Update version to 0.5.2 for pypi release.
|
3 years ago |