Added support for Kaunet loss replay