From 4d5ff32597a8e7e4fa9eae8c1b5e54f3e93fa76f Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Mon, 17 Mar 2025 11:21:43 +0100
Subject: [PATCH] add torch version in requirements.txt

---
 requirements.txt | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index b466db9..26d4c28 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,6 +11,12 @@ PyYAML>=5.3.1        # YAML parser and emitter
 pandas==2.0.3        # Data manipulation and analysis
 tqdm>=4.64.0         # Progress bar library
 
+# Torch -----------------------------------------------------------------------
+
+torch==2.0.0        
+torchvision==0.15.1
+torchaudio==2.0.1
+
 # Plotting --------------------------------------------------------------------
 
 seaborn>=0.11.0      # Statistical data visualization
-- 
GitLab