Commit Graph

3 Commits (eccbadca740579857f4a76bd67b3d6f29290913d)

Author SHA1 Message Date
Ross Wightman 3b8f63084d Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way 2019-11-28 17:04:35 -08:00
Ross Wightman 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
Ross Wightman 8a33a6c90a Add checkpoint clean script, add link to pretrained resnext50 weights 2019-04-13 14:15:35 -07:00