keras-swa (pypi): Simple stochastic weight averaging callback for Keras.
Keywords: deep-learning|keras|stochastic-weight-averaging|swa|tensorflow|tensorflow-keras