Skip to content
Snippets Groups Projects
Commit 720dd3ce authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

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

parent 8f8b627f
No related branches found
No related tags found
1 merge request!6autoencoders pretraining using RGB faces
Pipeline #26174 failed
...@@ -36,11 +36,6 @@ requirements: ...@@ -36,11 +36,6 @@ requirements:
- setuptools - setuptools
- numpy >=1.11 - numpy >=1.11
- docopt - docopt
- os
- argparse
- importlib
- logging
- time
- pytorch =0.4 - pytorch =0.4
- torchvision >=0.2.0 - torchvision >=0.2.0
- matplotlib - matplotlib
......
...@@ -8,7 +8,3 @@ bob.io.base ...@@ -8,7 +8,3 @@ bob.io.base
bob.io.image bob.io.image
torch >= 0.4.0 torch >= 0.4.0
torchvision >= 0.2.0 torchvision >= 0.2.0
argparse
importlib
logging
time
...@@ -22,7 +22,7 @@ setup( ...@@ -22,7 +22,7 @@ setup(
author = 'Guillaume Heusch', author = 'Guillaume Heusch',
author_email = 'guillaume.heusch@idiap.ch', author_email = 'guillaume.heusch@idiap.ch',
keywords = 'pytorch', keywords = 'pytorch',
# If you have a better, long description of your package, place it on the # If you have a better, long description of your package, place it on the
# 'doc' directory and then hook it here # 'doc' directory and then hook it here
long_description = open('README.rst').read(), long_description = open('README.rst').read(),
...@@ -69,9 +69,10 @@ setup( ...@@ -69,9 +69,10 @@ setup(
# scripts should be declared using this entry: # scripts should be declared using this entry:
'console_scripts' : [ 'console_scripts' : [
'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main', 'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main',
'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main', 'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main',
'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main', 'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main',
'pytorch-train-autoencoder-pad.py = bob.pad.face.script.pytorch.pytorch_train:main',
], ],
}, },
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment