diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-21 20:03:10 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-03-21 20:03:10 +0100 |
commit | aac452a2dc008338cb543549652da293c14b6b4e (patch) | |
tree | 6d018841e28f22eee5289f6cc59c28100a9d023d /pyproject.toml | |
parent | a3a40c9c0118039460d5c9fba6a74edc0cdba106 (diff) |
Refactor EMNIST dataset
Diffstat (limited to 'pyproject.toml')
-rw-r--r-- | pyproject.toml | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml index 2f774b2..ef75edf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -40,6 +40,7 @@ wandb = "^0.10.12" einops = "^0.3.0" gtn = "^0.0.0" sentencepiece = "^0.1.95" +pytorch-lightning = "^1.2.4" [tool.poetry.dev-dependencies] pytest = "^5.4.2" |