log GPU memory usage

2023-08-09 08:10:37 +00:00
parent 176b888a63
commit e303d64728
6 changed files with 63 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,3 +19,4 @@ evaluate==0.4.0
 rouge-score==0.1.2
 scipy
 scikit-learn==1.2.2
+nvidia-ml-py3