add torch dependency

Coda-Research-Group · Jan 31, 2025 · 748a3aa · 748a3aa
1 parent fe53af2
commit 748a3aa
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/training/requirements.txt b/training/requirements.txt
@@ -1,10 +1,9 @@
 pandas
 pillow
-keras
+torch
 scikit-learn
 matplotlib
 numpy
-pandas
 pyyaml
 kaleido
 faiss-cpu