Add torchvision dependency.

This commit is contained in:
David Bau
2022-06-01 09:53:43 -04:00
parent 887011a28e
commit 78b593c137
+1
View File
@@ -22,3 +22,4 @@ python_requires = >=3.7
install_requires =
numpy
torch
torchvision