diff --git a/.gitattributes b/.gitattributes
index fc14369f2a62dac1c6ce24706c549137730bc6a2..b94deb084b48af8a2c148dde6665466ee91acb39 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -1540,3 +1540,28 @@ generated_samples/epoch_500/cfg_1/class_1_preview.png filter=lfs diff=lfs merge=
 generated_samples/epoch_500/cfg_1/class_2_preview.png filter=lfs diff=lfs merge=lfs -text
 generated_samples/epoch_500/cfg_1/class_3_preview.png filter=lfs diff=lfs merge=lfs -text
 generated_samples/epoch_500/cfg_1/class_4_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_1/class_5_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_1/class_6_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_1/class_7_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_1/class_8_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_1/class_9_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_0_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_1_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_2_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_3_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_4_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_5_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_6_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_7_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_8_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_2/class_9_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_0_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_1_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_2_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_3_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_4_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_5_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_6_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_7_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_8_preview.png filter=lfs diff=lfs merge=lfs -text
+generated_samples/epoch_500/cfg_3/class_9_preview.png filter=lfs diff=lfs merge=lfs -text
diff --git a/epoch_500/cfg_1/class_5.pt b/generated_samples/epoch_500/cfg_1/class_5.pt
similarity index 100%
rename from epoch_500/cfg_1/class_5.pt
rename to generated_samples/epoch_500/cfg_1/class_5.pt
diff --git a/epoch_500/cfg_1/class_5_preview.png b/generated_samples/epoch_500/cfg_1/class_5_preview.png
similarity index 100%
rename from epoch_500/cfg_1/class_5_preview.png
rename to generated_samples/epoch_500/cfg_1/class_5_preview.png
diff --git a/epoch_500/cfg_1/class_6.pt b/generated_samples/epoch_500/cfg_1/class_6.pt
similarity index 100%
rename from epoch_500/cfg_1/class_6.pt
rename to generated_samples/epoch_500/cfg_1/class_6.pt
diff --git a/epoch_500/cfg_1/class_6_preview.png b/generated_samples/epoch_500/cfg_1/class_6_preview.png
similarity index 100%
rename from epoch_500/cfg_1/class_6_preview.png
rename to generated_samples/epoch_500/cfg_1/class_6_preview.png
diff --git a/epoch_500/cfg_1/class_7.pt b/generated_samples/epoch_500/cfg_1/class_7.pt
similarity index 100%
rename from epoch_500/cfg_1/class_7.pt
rename to generated_samples/epoch_500/cfg_1/class_7.pt
diff --git a/epoch_500/cfg_1/class_7_preview.png b/generated_samples/epoch_500/cfg_1/class_7_preview.png
similarity index 100%
rename from epoch_500/cfg_1/class_7_preview.png
rename to generated_samples/epoch_500/cfg_1/class_7_preview.png
diff --git a/epoch_500/cfg_1/class_8.pt b/generated_samples/epoch_500/cfg_1/class_8.pt
similarity index 100%
rename from epoch_500/cfg_1/class_8.pt
rename to generated_samples/epoch_500/cfg_1/class_8.pt
diff --git a/epoch_500/cfg_1/class_8_preview.png b/generated_samples/epoch_500/cfg_1/class_8_preview.png
similarity index 100%
rename from epoch_500/cfg_1/class_8_preview.png
rename to generated_samples/epoch_500/cfg_1/class_8_preview.png
diff --git a/epoch_500/cfg_1/class_9.pt b/generated_samples/epoch_500/cfg_1/class_9.pt
similarity index 100%
rename from epoch_500/cfg_1/class_9.pt
rename to generated_samples/epoch_500/cfg_1/class_9.pt
diff --git a/epoch_500/cfg_1/class_9_preview.png b/generated_samples/epoch_500/cfg_1/class_9_preview.png
similarity index 100%
rename from epoch_500/cfg_1/class_9_preview.png
rename to generated_samples/epoch_500/cfg_1/class_9_preview.png
diff --git a/epoch_500/cfg_2/class_0.pt b/generated_samples/epoch_500/cfg_2/class_0.pt
similarity index 100%
rename from epoch_500/cfg_2/class_0.pt
rename to generated_samples/epoch_500/cfg_2/class_0.pt
diff --git a/epoch_500/cfg_2/class_0_preview.png b/generated_samples/epoch_500/cfg_2/class_0_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_0_preview.png
rename to generated_samples/epoch_500/cfg_2/class_0_preview.png
diff --git a/epoch_500/cfg_2/class_1.pt b/generated_samples/epoch_500/cfg_2/class_1.pt
similarity index 100%
rename from epoch_500/cfg_2/class_1.pt
rename to generated_samples/epoch_500/cfg_2/class_1.pt
diff --git a/epoch_500/cfg_2/class_1_preview.png b/generated_samples/epoch_500/cfg_2/class_1_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_1_preview.png
rename to generated_samples/epoch_500/cfg_2/class_1_preview.png
diff --git a/epoch_500/cfg_2/class_2.pt b/generated_samples/epoch_500/cfg_2/class_2.pt
similarity index 100%
rename from epoch_500/cfg_2/class_2.pt
rename to generated_samples/epoch_500/cfg_2/class_2.pt
diff --git a/epoch_500/cfg_2/class_2_preview.png b/generated_samples/epoch_500/cfg_2/class_2_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_2_preview.png
rename to generated_samples/epoch_500/cfg_2/class_2_preview.png
diff --git a/epoch_500/cfg_2/class_3.pt b/generated_samples/epoch_500/cfg_2/class_3.pt
similarity index 100%
rename from epoch_500/cfg_2/class_3.pt
rename to generated_samples/epoch_500/cfg_2/class_3.pt
diff --git a/epoch_500/cfg_2/class_3_preview.png b/generated_samples/epoch_500/cfg_2/class_3_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_3_preview.png
rename to generated_samples/epoch_500/cfg_2/class_3_preview.png
diff --git a/epoch_500/cfg_2/class_4.pt b/generated_samples/epoch_500/cfg_2/class_4.pt
similarity index 100%
rename from epoch_500/cfg_2/class_4.pt
rename to generated_samples/epoch_500/cfg_2/class_4.pt
diff --git a/epoch_500/cfg_2/class_4_preview.png b/generated_samples/epoch_500/cfg_2/class_4_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_4_preview.png
rename to generated_samples/epoch_500/cfg_2/class_4_preview.png
diff --git a/epoch_500/cfg_2/class_5.pt b/generated_samples/epoch_500/cfg_2/class_5.pt
similarity index 100%
rename from epoch_500/cfg_2/class_5.pt
rename to generated_samples/epoch_500/cfg_2/class_5.pt
diff --git a/epoch_500/cfg_2/class_5_preview.png b/generated_samples/epoch_500/cfg_2/class_5_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_5_preview.png
rename to generated_samples/epoch_500/cfg_2/class_5_preview.png
diff --git a/epoch_500/cfg_2/class_6.pt b/generated_samples/epoch_500/cfg_2/class_6.pt
similarity index 100%
rename from epoch_500/cfg_2/class_6.pt
rename to generated_samples/epoch_500/cfg_2/class_6.pt
diff --git a/epoch_500/cfg_2/class_6_preview.png b/generated_samples/epoch_500/cfg_2/class_6_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_6_preview.png
rename to generated_samples/epoch_500/cfg_2/class_6_preview.png
diff --git a/epoch_500/cfg_2/class_7.pt b/generated_samples/epoch_500/cfg_2/class_7.pt
similarity index 100%
rename from epoch_500/cfg_2/class_7.pt
rename to generated_samples/epoch_500/cfg_2/class_7.pt
diff --git a/epoch_500/cfg_2/class_7_preview.png b/generated_samples/epoch_500/cfg_2/class_7_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_7_preview.png
rename to generated_samples/epoch_500/cfg_2/class_7_preview.png
diff --git a/epoch_500/cfg_2/class_8.pt b/generated_samples/epoch_500/cfg_2/class_8.pt
similarity index 100%
rename from epoch_500/cfg_2/class_8.pt
rename to generated_samples/epoch_500/cfg_2/class_8.pt
diff --git a/epoch_500/cfg_2/class_8_preview.png b/generated_samples/epoch_500/cfg_2/class_8_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_8_preview.png
rename to generated_samples/epoch_500/cfg_2/class_8_preview.png
diff --git a/epoch_500/cfg_2/class_9.pt b/generated_samples/epoch_500/cfg_2/class_9.pt
similarity index 100%
rename from epoch_500/cfg_2/class_9.pt
rename to generated_samples/epoch_500/cfg_2/class_9.pt
diff --git a/epoch_500/cfg_2/class_9_preview.png b/generated_samples/epoch_500/cfg_2/class_9_preview.png
similarity index 100%
rename from epoch_500/cfg_2/class_9_preview.png
rename to generated_samples/epoch_500/cfg_2/class_9_preview.png
diff --git a/epoch_500/cfg_3/class_0.pt b/generated_samples/epoch_500/cfg_3/class_0.pt
similarity index 100%
rename from epoch_500/cfg_3/class_0.pt
rename to generated_samples/epoch_500/cfg_3/class_0.pt
diff --git a/epoch_500/cfg_3/class_0_preview.png b/generated_samples/epoch_500/cfg_3/class_0_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_0_preview.png
rename to generated_samples/epoch_500/cfg_3/class_0_preview.png
diff --git a/epoch_500/cfg_3/class_1.pt b/generated_samples/epoch_500/cfg_3/class_1.pt
similarity index 100%
rename from epoch_500/cfg_3/class_1.pt
rename to generated_samples/epoch_500/cfg_3/class_1.pt
diff --git a/epoch_500/cfg_3/class_1_preview.png b/generated_samples/epoch_500/cfg_3/class_1_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_1_preview.png
rename to generated_samples/epoch_500/cfg_3/class_1_preview.png
diff --git a/epoch_500/cfg_3/class_2.pt b/generated_samples/epoch_500/cfg_3/class_2.pt
similarity index 100%
rename from epoch_500/cfg_3/class_2.pt
rename to generated_samples/epoch_500/cfg_3/class_2.pt
diff --git a/epoch_500/cfg_3/class_2_preview.png b/generated_samples/epoch_500/cfg_3/class_2_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_2_preview.png
rename to generated_samples/epoch_500/cfg_3/class_2_preview.png
diff --git a/epoch_500/cfg_3/class_3.pt b/generated_samples/epoch_500/cfg_3/class_3.pt
similarity index 100%
rename from epoch_500/cfg_3/class_3.pt
rename to generated_samples/epoch_500/cfg_3/class_3.pt
diff --git a/epoch_500/cfg_3/class_3_preview.png b/generated_samples/epoch_500/cfg_3/class_3_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_3_preview.png
rename to generated_samples/epoch_500/cfg_3/class_3_preview.png
diff --git a/epoch_500/cfg_3/class_4.pt b/generated_samples/epoch_500/cfg_3/class_4.pt
similarity index 100%
rename from epoch_500/cfg_3/class_4.pt
rename to generated_samples/epoch_500/cfg_3/class_4.pt
diff --git a/epoch_500/cfg_3/class_4_preview.png b/generated_samples/epoch_500/cfg_3/class_4_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_4_preview.png
rename to generated_samples/epoch_500/cfg_3/class_4_preview.png
diff --git a/epoch_500/cfg_3/class_5.pt b/generated_samples/epoch_500/cfg_3/class_5.pt
similarity index 100%
rename from epoch_500/cfg_3/class_5.pt
rename to generated_samples/epoch_500/cfg_3/class_5.pt
diff --git a/epoch_500/cfg_3/class_5_preview.png b/generated_samples/epoch_500/cfg_3/class_5_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_5_preview.png
rename to generated_samples/epoch_500/cfg_3/class_5_preview.png
diff --git a/epoch_500/cfg_3/class_6.pt b/generated_samples/epoch_500/cfg_3/class_6.pt
similarity index 100%
rename from epoch_500/cfg_3/class_6.pt
rename to generated_samples/epoch_500/cfg_3/class_6.pt
diff --git a/epoch_500/cfg_3/class_6_preview.png b/generated_samples/epoch_500/cfg_3/class_6_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_6_preview.png
rename to generated_samples/epoch_500/cfg_3/class_6_preview.png
diff --git a/epoch_500/cfg_3/class_7.pt b/generated_samples/epoch_500/cfg_3/class_7.pt
similarity index 100%
rename from epoch_500/cfg_3/class_7.pt
rename to generated_samples/epoch_500/cfg_3/class_7.pt
diff --git a/epoch_500/cfg_3/class_7_preview.png b/generated_samples/epoch_500/cfg_3/class_7_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_7_preview.png
rename to generated_samples/epoch_500/cfg_3/class_7_preview.png
diff --git a/epoch_500/cfg_3/class_8.pt b/generated_samples/epoch_500/cfg_3/class_8.pt
similarity index 100%
rename from epoch_500/cfg_3/class_8.pt
rename to generated_samples/epoch_500/cfg_3/class_8.pt
diff --git a/epoch_500/cfg_3/class_8_preview.png b/generated_samples/epoch_500/cfg_3/class_8_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_8_preview.png
rename to generated_samples/epoch_500/cfg_3/class_8_preview.png
diff --git a/epoch_500/cfg_3/class_9.pt b/generated_samples/epoch_500/cfg_3/class_9.pt
similarity index 100%
rename from epoch_500/cfg_3/class_9.pt
rename to generated_samples/epoch_500/cfg_3/class_9.pt
diff --git a/epoch_500/cfg_3/class_9_preview.png b/generated_samples/epoch_500/cfg_3/class_9_preview.png
similarity index 100%
rename from epoch_500/cfg_3/class_9_preview.png
rename to generated_samples/epoch_500/cfg_3/class_9_preview.png