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		6 years ago
..
__init__.py	…
adam_tf.py	…
nadam.py	…
optim_factory.py	…
rmsprop_tf.py	Add exponential moving average for model weights + few other additions and cleanup	6 years ago