Skip to content
This repository has been archived by the owner on Jul 16, 2021. It is now read-only.

Adam Optimizer implementation #213

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Conversation

sixilli
Copy link

@sixilli sixilli commented Jan 29, 2020

This is my first pull request ever so hopefully it's not too far off from usable. I'm having some difficulties with testing my code. I plugged in my adam optimizer to the example and the trained network guessed 0.0 four times. While the default code within the example folder for a MLP worked as planned.

For the overall structuring of the code I did my best to follow the RMSPROP optimizer.

EDIT: Fixed some math errors and it seems to operate as expected now. Both the cost is decreasing and it correctly passed the example test. I also improved the memory usage of the previous version by using the in place util.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant