Commit 16de4a67 authored by Anjith GEORGE's avatar Anjith GEORGE Committed by Anjith GEORGE
Browse files

Changed torch version

parent f2da5438
......@@ -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
......
......@@ -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
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment