Update README.md

main
PENG Bo 4 years ago committed by GitHub
parent 619c8add7a
commit 6b1ba8a9bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -20,6 +20,8 @@ kv / k is the memory mechanism. The token with high k can be remembered for a lo
It's also using my SmallInitEmb trick https://github.com/BlinkDL/SmallInitEmb (applicable to all transformers), and a custom CUDA kernel https://github.com/BlinkDL/RWKV-CUDA .
I find it might be nice to make the model stay on a mid-lr for a long period, because in theory that's where most learning shall happen. For example: 6e-4 to 1e-4 in 15% of steps, stays on 1e-4 for 60% of steps, then 1e-4 to 1e-5 in 25% of steps.
The pseudocode (execution from top to bottom):
![RWKV-v2-RNN](RWKV-v2-RNN.png)

Loading…
Cancel
Save