pytorch-image-models

Author	SHA1	Message	Date
Ross Wightman	0e1fd11ad8	Merge pull request #12 from rwightman/ema-cleanup Model weights Exponential Moving Average	5 years ago
Ross Wightman	1019414fd2	Add ported Tensorflow EfficientNet B4/B5 weights	5 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	5 years ago
Ross Wightman	0562b91c38	Add per model crop pct, interpolation defaults, tie it all together * create one resolve fn to pull together model defaults + cmd line args * update attribution comments in some models * test update train/validation/inference scripts	6 years ago
Ross Wightman	9c3859fb9c	Uniform pretrained model handling. * All models have 'default_cfgs' dict * load/resume/pretrained helpers factored out * pretrained load operates on state_dict based on default_cfg * test all models in validate * schedule, optim factor factored out * test time pool wrapper applied based on default_cfg	6 years ago