adding tensorboard writer for training loss and cv loss

2021-07-26 02:38:50 -07:00
parent 969812c5cc
commit 7b2ca6edc7
3 changed files with 17 additions and 6 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ torch
 sklearn
 pytest
 json5
-tqdm
+tqdm
+tensorboard