From 720dd3ce86490bb5855d944d8106396a00b58739 Mon Sep 17 00:00:00 2001 From: Olegs NIKISINS <onikisins@italix03.idiap.ch> Date: Mon, 21 Jan 2019 15:32:29 +0100 Subject: [PATCH] Removed standard lib packages from meta.yaml, added entry point for training script --- conda/meta.yaml | 5 ----- requirements.txt | 4 ---- setup.py | 9 +++++---- 3 files changed, 5 insertions(+), 13 deletions(-) diff --git a/conda/meta.yaml b/conda/meta.yaml index 3aef6e6..29bd530 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 0181f38..c90de73 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 0161db0..56fad4b 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', ], }, -- GitLab