Commit 720dd3ce authored by Olegs NIKISINS's avatar Olegs NIKISINS

Removed standard lib packages from meta.yaml, added entry point for training script

parent 8f8b627f
Pipeline #26174 failed with stage
in 7 minutes and 47 seconds
......@@ -36,11 +36,6 @@ requirements:
- setuptools
- numpy >=1.11
- docopt
- os
- argparse
- importlib
- logging
- time
- pytorch =0.4
- torchvision >=0.2.0
- matplotlib
......
......@@ -22,7 +22,7 @@ setup(
author = 'Guillaume Heusch',
author_email = 'guillaume.heusch@idiap.ch',
keywords = 'pytorch',
# If you have a better, long description of your package, place it on the
# 'doc' directory and then hook it here
long_description = open('README.rst').read(),
......@@ -69,9 +69,10 @@ setup(
# scripts should be declared using this entry:
'console_scripts' : [
'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main',
'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main',
'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main',
'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main',
'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main',
'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main',
'pytorch-train-autoencoder-pad.py = bob.pad.face.script.pytorch.pytorch_train:main',
],
},
......
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