Refectored restoring and EMA bug fix
Refectored restoring and EMA bug fix

Fixed bug that EMAs have not been restored when doing a clean restore from a checkpoint file.

bsautermeister/tensorlight,bsautermeister/tensorlight