We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b8ebd7c commit 7fe523aCopy full SHA for 7fe523a
requirements.txt
@@ -1,7 +1,6 @@
1
# --------- pytorch --------- #
2
torch==1.8.1
3
-torchvision==0.9.1
4
-pytorch-lightning>=1.3.2
+pytorch-lightning==1.3.8
5
torchmetrics>=0.3.2
6
onnxruntime
7
onnxruntime-gpu
0 commit comments