Thank you for the code. This seems to me nicer than author's original PyTorch implementation. Can you please extend it with the meSimp method which was also introduced by same authors (check the link).
https://github.com/lancopku/meProp
https://github.com/lancopku/meProp/tree/master/src/pytorch
Thank you for the code. This seems to me nicer than author's original PyTorch implementation. Can you please extend it with the meSimp method which was also introduced by same authors (check the link).
https://github.com/lancopku/meProp
https://github.com/lancopku/meProp/tree/master/src/pytorch