Commit Graph

10 Commits (2ee398d501c708e8b117615056ff7d94f96aece7)

Author SHA1 Message Date
Ross Wightman 40457e5691 Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing...
3 years ago
Csaba Kertesz 5114c214fc Change the Python interpreter to Python 3.x in the scripts
4 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
5 years ago
Ross Wightman 40fea63ebe Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
Ross Wightman 3bef524f9c Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
5 years ago
Ross Wightman 3b8f63084d Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way
5 years ago
Ross Wightman 9bcd65181b Add exponential moving average for model weights + few other additions and cleanup
6 years ago
Ross Wightman 8a33a6c90a Add checkpoint clean script, add link to pretrained resnext50 weights
6 years ago