diff --git a/.gitignore b/.gitignore
index 25ea1ae..82ec14c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
-*/train_dir
+*/dataset_dir
+*/env
+*/__pycache__