Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
1743 Commits
41 Branches
41 Tags
28 MiB
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'main'
${ noResults }
Commit Graph

15 Commits (main)

Author SHA1 Message Date
Ross Wightman d0b45c9b4d Make safetensor import option for now. Improve avg/clean checkpoints ext handling a bit (more consistent).
1 year ago
testbot 8470e29541 Add support to load safetensors weights
1 year ago
Ross Wightman 927f031293 Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
1 year ago
Ross Wightman abc9ba2544 Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2 years ago
Ross Wightman 38804c721b Checkpoint clean fn useable stand alone
3 years ago
Ross Wightman ae1ff5792f Clean a1/a2/3 rsb _0 checkpoints properly, fix v2 loading.
3 years ago
Csaba Kertesz 5114c214fc Change the Python interpreter to Python 3.x in the scripts
3 years ago
Ross Wightman 9305313291 Default to old checkpoint format for now, still want compatibility with older torch ver for released models
4 years ago
Ross Wightman 6c17d57a2c Fix some attributions, add copyrights to some file docstrings
4 years ago
Ross Wightman cc0b1f4130 Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
4 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
4 years ago
Ross Wightman 3bef524f9c Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
4 years ago
Ross Wightman 3b8f63084d Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way
4 years ago
Ross Wightman 9bcd65181b Add exponential moving average for model weights + few other additions and cleanup
5 years ago
Ross Wightman 8a33a6c90a Add checkpoint clean script, add link to pretrained resnext50 weights
5 years ago