Skip to content

Commit 5e936a3

Browse files
committed
diagram
1 parent 9f0fa12 commit 5e936a3

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
<img src="./grokfast.png" width="400px"></img>
2+
13
## Grokfast - Pytorch (wip)
24

35
Explorations into <a href="https://arxiv.org/html/2405.20233v2">"Grokfast, Accelerated Grokking by Amplifying Slow Gradients"</a>, out of Seoul National University in Korea. In particular, will compare it with NAdam on modular addition as well as a few other tasks, since I am curious why those experiments are left out of the paper. If it holds up, will polish it up into a nice package for quick use.

grokfast.png

142 KB
Loading

0 commit comments

Comments
 (0)