Update README.md

Jiayi-Pan · Feb 1, 2025 · 8a62392 · 8a62392
1 parent 4298ed3
commit 8a62392
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -45,7 +45,7 @@ python ./examples/data_preprocess/countdown.py --local_dir {path_to_your_dataset
 conda activate zero
 ```
 
-For the following code, if you see Out-of-vram, try add `critic.model.enable_gradient_checkpointing=True` to the script
+For the following code, if you see Out-of-vram, try add `critic.model.enable_gradient_checkpointing=True` to the script, and checkout the discussion [here](https://github.com/Jiayi-Pan/TinyZero/issues/5#issuecomment-2624161643)
 
 **Single GPU**