diff --git a/conda/meta.yaml b/conda/meta.yaml index 3aef6e639dcf23e7afa6532d21b22106bf047787..29bd5302dce9c4134e9002a521fdaa75092fde58 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -36,11 +36,6 @@ requirements: - setuptools - numpy >=1.11 - docopt - - os - - argparse - - importlib - - logging - - time - pytorch =0.4 - torchvision >=0.2.0 - matplotlib diff --git a/requirements.txt b/requirements.txt index 0181f38f51a40c73f5ab2b0b2e505c9e660512dd..c90de733a190c9a2260ec8cbf9144c330ae25667 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,7 +8,3 @@ bob.io.base bob.io.image torch >= 0.4.0 torchvision >= 0.2.0 -argparse -importlib -logging -time diff --git a/setup.py b/setup.py index 0161db0e6071f8922bd54ff75b2d10dd35e8b1c9..56fad4be48222ae335ae179e3ea617a7d08598fc 100644 --- a/setup.py +++ b/setup.py @@ -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', ], },