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