Add torchviz and notebook for visualisation

2024-01-23 14:51:02 +01:00 · 2024-01-23 14:51:02 +01:00 · a8ce121f5f
commit a8ce121f5f
parent b79ea3178d
1 changed files with 2 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,3 +1,5 @@
 torch
 torchvision
 lmdb
+torchviz
+notebook