Update README.md

4 years ago · e541d93b97
parent 6b1ba8a9bd
commit e541d93b97
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -20,7 +20,7 @@ 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.
+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 (actually I monitor the loss and decay the lr when it plateaus), then 1e-4 to 1e-5 in 25% of steps.

 The pseudocode (execution from top to bottom):