Skip to content

Commit 592dce5

Browse files
danielstapsdanielstaps
committed
Pin the prototorch/torch stack to verified versions (reproducibility)
prototorch==0.7.6, prototorch_models==0.7.1, torchmetrics>=1.0 (task= API), torch>=2.0 — the versions verified working in-container.
1 parent d842e24 commit 592dce5

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎requirements.txt‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
numpy
22
pandas
33
matplotlib
4-
torch
5-
prototorch
6-
prototorch_models
4+
torch>=2.0
5+
prototorch==0.7.6
6+
prototorch_models==0.7.1
77
# optional, only for the tensorflow-datasets loaders:
88
# tensorflow
99
# tensorflow-datasets

0 commit comments

Comments
 (0)