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
6635bc3f7d
Merge pull request #1479 from rwightman/script_cleanup
Train / val script enhancements, non-GPU (ie CPU) device support, HF datasets support, TFDS/WDS dataloading improvements
|
2 years ago |
| .. |
|
data
|
Merge remote-tracking branch 'origin/master' into script_cleanup
|
2 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
|
Merge pull request #1479 from rwightman/script_cleanup
|
2 years ago |
|
optim
|
Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
|
2 years ago |
|
scheduler
|
Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
|
2 years ago |
|
utils
|
Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
|
2 years ago |
|
__init__.py
|
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
|
3 years ago |
|
version.py
|
Version 0.7.dev0 for master
|
2 years ago |
