add torch version in requirements.txt

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