From aac452a2dc008338cb543549652da293c14b6b4e Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 21 Mar 2021 20:03:10 +0100
Subject: Refactor EMNIST dataset

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

(limited to 'pyproject.toml')

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"
-- 
cgit v1.2.3-70-g09d2