summaryrefslogtreecommitdiff
path: root/pyproject.toml
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-11-08 14:54:44 +0100
committeraktersnurra <gustaf.rydholm@gmail.com>2020-11-08 14:54:44 +0100
commitdc28cbe2b4ed77be92ee8b2b69a20689c3bf02a4 (patch)
tree1b5fc0d06952e13727e85c4f973a26d277068453 /pyproject.toml
parente181195a699d7fa237f256d90ab4dedffc03d405 (diff)
new updates
Diffstat (limited to 'pyproject.toml')
-rw-r--r--pyproject.toml16
1 files changed, 11 insertions, 5 deletions
diff --git a/pyproject.toml b/pyproject.toml
index 43943e7..95e32a6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,6 +35,10 @@ wandb = "^0.9.6"
torch-summary = "^1.4.2"
python-Levenshtein = "^0.12.0"
defusedxml = "^0.6.0"
+pytorch-block-sparse = "^0.1.2"
+pytorch-metric-learning = "^0.9.92"
+faiss-gpu = "^1.6.3"
+omegaconf = "^2.0.2"
[tool.poetry.dev-dependencies]
pytest = "^5.4.2"
@@ -62,12 +66,14 @@ wandb = "^0.9.4"
fail_under = 50
[tool.poetry.scripts]
-download-emnist = "text_recognizer.datasets.emnist_dataset:download_emnist"
+download-emnist = "text_recognizer.datasets.util:download_emnist"
+download-iam = "text_recognizer.datasets.iam_dataset:main"
create-emnist-support-files = "text_recognizer.tests.support.create_emnist_support_files:create_emnist_support_files"
-run-experiment = "training.run_experiment:main"
-run-experiments = "training.prepare_experiments:main"
-# mlp = "text_recognizer.networks.mlp:test"
-# lenet = "text_recognizer.networks.lenet:test"
+create-emnist-lines-datasets = "text_recognizer.datasets.emnist_lines_dataset:create_datasets"
+create-iam-paragraphs = "text_recognizer.datasets.iam_paragraphs_dataset:main"
+prepare-experiments = "training.prepare_experiments:run_cli"
+
+
[build-system]
requires = ["poetry>=0.12"]