You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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
..
__init__.py	Remove adabound optimizer, never got it working well on larger datasets	6 years ago
adam_tf.py	MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def	6 years ago
nadam.py	Update a few comment, add some references	6 years ago
optim_factory.py	Remove adabound optimizer, never got it working well on larger datasets	6 years ago
rmsprop_tf.py	Add exponential moving average for model weights + few other additions and cleanup	5 years ago