diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-21 22:33:58 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-21 22:33:58 +0100 |
| commit | e3741de333a3a43a7968241b6eccaaac66dd7b20 (patch) | |
| tree | 7c50aee4ca61f77e95f1b038030292c64bbb86c2 /pyproject.toml | |
| parent | aac452a2dc008338cb543549652da293c14b6b4e (diff) | |
Working on EMNIST Lines dataset
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/pyproject.toml b/pyproject.toml index ef75edf..33d539e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,8 +22,8 @@ sphinx_rtd_theme = "^0.4.3" boltons = "^20.1.0" h5py = "^3.2.1" toml = "^0.10.1" -torch = "^1.7.0" -torchvision = "^0.8.1" +torch = "^1.8.0" +torchvision = "^0.9.0" loguru = "^0.5.0" matplotlib = "^3.2.1" tqdm = "^4.46.1" @@ -64,12 +64,13 @@ gpustat = "^0.6.0" redlock-py = "^1.0.8" wandb = "^0.10.11" graphviz = "^0.16" +scipy = "^1.6.1" [tool.coverage.report] fail_under = 50 [tool.poetry.scripts] -download-emnist = "text_recognizer.datasets.util:download_emnist" +download-emnist = "text_recognizer.datasets.emnist: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" create-emnist-lines-datasets = "text_recognizer.datasets.emnist_lines_dataset:create_datasets" @@ -78,7 +79,6 @@ prepare-experiments = "training.prepare_experiments:run_cli" run-experiment = "training.run_experiment:run_cli" - [build-system] requires = ["poetry>=0.12"] build-backend = "poetry.masonry.api" |