d0b45c9b4d
Make safetensor import option for now. Improve avg/clean checkpoints ext handling a bit (more consistent).
2023-02-18 16:06:42 -08:00
8470e29541
Add support to load safetensors weights
2023-02-16 16:20:31 -08:00
927f031293
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
2022-12-06 15:00:06 -08:00
abc9ba2544
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2022-01-25 21:54:13 -08:00
38804c721b
Checkpoint clean fn useable stand alone
2021-10-08 17:43:53 -07:00
ae1ff5792f
Clean a1/a2/3 rsb _0 checkpoints properly, fix v2 loading.
2021-10-04 16:46:00 -07:00
5114c214fc
Change the Python interpreter to Python 3.x in the scripts
2021-02-09 21:20:28 +02:00
9305313291
Default to old checkpoint format for now, still want compatibility with older torch ver for released models
2020-10-13 12:58:04 -07:00
6c17d57a2c
Fix some attributions, add copyrights to some file docstrings
2020-07-27 13:44:56 -07:00
cc0b1f4130
Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm
2020-01-12 17:52:19 -08:00
40fea63ebe
Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
2020-01-03 14:57:46 -08:00
3bef524f9c
Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
2019-11-29 17:56:36 -08:00
3b8f63084d
Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way
2019-11-28 17:04:35 -08:00
9bcd65181b
Add exponential moving average for model weights + few other additions and cleanup
...
* ModelEma class added to track an EMA set of weights for the model being trained
* EMA handling added to train, validation and clean_checkpoint scripts
* Add multi checkpoint or multi-model validation support to validate.py
* Add syncbn option (APEX) to train script for experimentation
* Cleanup interface of CheckpointSaver while adding ema functionality
2019-06-07 15:39:36 -07:00
8a33a6c90a
Add checkpoint clean script, add link to pretrained resnext50 weights
2019-04-13 14:15:35 -07:00
Ross Wightman
testbot
Csaba Kertesz