7 Commits (bf865d32c267346fd03b198c54eee2de74167190)

Author	SHA1	Message	Date
Vyacheslav Shults	44ed0f85e8	Refactor imports and indents according to PEP. ... Add proposal for links to pretrained weights. Unify style of reset_classifier function. Replace None by nn.Identity() in torch modules (all empty classifiers and HRNet)	5 years ago
Ross Wightman	cc0b1f4130	Add support to clean_checkpoint.py to remove aux_bn weights/biases from SplitBatchNorm	6 years ago
Ross Wightman	40fea63ebe	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	6 years ago
Ross Wightman	3bef524f9c	Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.	6 years ago
Ross Wightman	3b8f63084d	Clean checkpoint renames pth w/ SHA hash in a torch hub compatible way	6 years ago
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	6 years ago
Ross Wightman	8a33a6c90a	Add checkpoint clean script, add link to pretrained resnext50 weights	6 years ago