diff --git a/pyproject.toml b/pyproject.toml
index 0bc9e80..6a4305d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,6 +35,8 @@ torchvision = "^0.17.1"
 torchinfo = "^1.8.0"
 minio = "^7.2.7"
 tqdm = "^4.66.4"
+pytorch-ignite = "^0.5.1"
+tensorboard = "^2.17.1"
 
 
 [tool.poetry.group.shared.dependencies]