diff --git a/README.md b/README.md index 6fbb3ae..75bf961 100644 --- a/README.md +++ b/README.md @@ -25,7 +25,7 @@ See the release for a 27M params model on enwik8 with 0.72 BPC(dev). Run run.py Fine-tuning & training: https://github.com/BlinkDL/RWKV-LM/tree/main/RWKV-v2-RNN -Note: change 1e-15 to 1e-9 in https://github.com/BlinkDL/RWKV-LM/blob/main/RWKV-v2-RNN/src/model.py and https://github.com/BlinkDL/RWKV-LM/blob/main/RWKV-v2-RNN/src/model_run.py and probably you need other changes as well. You can compare the output with the latest code ( https://github.com/BlinkDL/RWKV-v2-RNN-Pile ) to verify it. +Note: For fine-tuning the Pile model, change 1e-15 to 1e-9 in https://github.com/BlinkDL/RWKV-LM/blob/main/RWKV-v2-RNN/src/model.py and https://github.com/BlinkDL/RWKV-LM/blob/main/RWKV-v2-RNN/src/model_run.py and probably you need other changes as well. You can compare the output with the latest code ( https://github.com/BlinkDL/RWKV-v2-RNN-Pile ) to verify it. ## How it works