diff --git a/README.md b/README.md
index 5003799..c198d09 100644
--- a/README.md
+++ b/README.md
@@ -212,6 +212,7 @@ This will store your a backup file with your current locally installed pip packa
     - Fix an issue without `.npz` and with `--full_path` in training.
     - Support extensions with upper cases for images for not Windows environment.
     - Fix `resize_lora.py` to work with LoRA with dynamic rank (including `conv_dim != network_dim`). Thanks to toshiaki!
+    - Fix issue: https://github.com/bmaltais/kohya_ss/issues/406
 * 2023/03/19 (v21.2.5):
     - Fix basic captioning logic
     - Add possibility to not train TE in Dreamboot by setting `Step text encoder training` to -1.
diff --git a/dreambooth_gui.py b/dreambooth_gui.py
index d7a0105..c3458be 100644
--- a/dreambooth_gui.py
+++ b/dreambooth_gui.py
@@ -331,11 +331,11 @@ def train_model(
     if check_if_model_exist(output_name, output_dir, save_model_as):
         return
 
-    # Get a list of all subfolders in train_data_dir
+    # Get a list of all subfolders in train_data_dir, excluding hidden folders
     subfolders = [
         f
         for f in os.listdir(train_data_dir)
-        if os.path.isdir(os.path.join(train_data_dir, f))
+        if os.path.isdir(os.path.join(train_data_dir, f)) and not f.startswith('.')
     ]
 
     total_steps = 0