Commit 49d611f5 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

switch back to using pytorch

parent 8b0e2ac0
Pipeline #29783 failed with stage
in 7 minutes and 9 seconds
......@@ -33,13 +33,13 @@ requirements:
- bob.bio.base
- bob.io.image
- bob.ap
- numpy
- docopt
- matplotlib
- h5py
- tensorflow
# - pytorch-cpu
# - torchvision-cpu
- numpy {{ numpy }}
- docopt {{ docopt }}
- matplotlib {{ matplotlib }}
- h5py {{ h5py }}
- tensorflow {{ tensorflow }}
- pytorch {{ pytorch }}
- torchvision {{ torchvision }}
run:
- python
- setuptools
......@@ -48,6 +48,8 @@ requirements:
- {{ pin_compatible('matplotlib') }}
- {{ pin_compatible('h5py') }}
- {{ pin_compatible('tensorflow') }}
- {{ pin_compatible('pytorch') }}
- {{ pin_compatible('torchvision') }}
test:
imports:
......@@ -69,8 +71,6 @@ test:
- coverage
- sphinx
- sphinx_rtd_theme
- pytorch-cpu
- torchvision-cpu
about:
home: https://www.idiap.ch/software/bob/
......
Markdown is supported
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