fixed torch version to 1.4 to prevent HPC issues

ca734793 · Andrei-Claudiu Roibu · 17bf6050 · ca734793
Commit ca734793 authored Apr 23, 2020 by Andrei-Claudiu Roibu 🖥
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@ setup(
        'nibabel',
        'numpy',
        'pandas',
-        'torch',
+        'torch==1.4',
        'h5py',
        'fslpy',
        'tensorboardX',