Changed torch version

conda/meta.yaml

@@ -10,6 +10,7 @@ build:
   entry_points:
     - train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main
     - train_mccnn.py = bob.learn.pytorch.scripts.train_mccnn:main
+    - train_generic.py = bob.learn.pytorch.scripts.train_generic:main
     - train_fasnet.py = bob.learn.pytorch.scripts.train_fasnet:main
     - train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main
     - train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main
@@ -57,6 +58,7 @@ test:
   commands:
     - train_cnn.py --help
     - train_mccnn.py --help
+    - train_generic.py --help
     - train_fasnet.py --help
     - train_dcgan.py --help
     - train_conditionalgan.py --help

requirements.txt

@@ -8,6 +8,6 @@ bob.io.base
 bob.bio.base
 bob.io.image
 bob.ap
-torch >= 0.4.1
-torchvision >= 0.2.1
+torch >=1.0.1
+torchvision >=0.2.1
 h5py