Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
|
@@ -2,4 +2,6 @@ torch>=2.0
|
|
| 2 |
torchvision
|
| 3 |
numpy
|
| 4 |
scikit-learn
|
| 5 |
-
tensorflow-datasets
|
|
|
|
|
|
|
|
|
| 2 |
torchvision
|
| 3 |
numpy
|
| 4 |
scikit-learn
|
| 5 |
+
tensorflow-datasets
|
| 6 |
+
onnx
|
| 7 |
+
onnxruntime # optional: for running/parity check
|