search

ebetica / autogradpp

Direct C++ Interface to PyTorch
MIT License
80 stars 12 forks source link

Adding amsgrad optimizer as an option for adam #39

Closed eugene-kharitonov closed 6 years ago

eugene-kharitonov commented 6 years ago

Following the code in pytorch