Preview of phase 2-5 implementation; needs a full check

2026-04-30 13:10:33 +01:00
parent 6e32001ebc
commit 7417267117
35 changed files with 3605 additions and 115 deletions
@@ -0,0 +1,10 @@
+{
+  "epochs": 100,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": true,
+  "image_size": 64,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000
+}
@@ -0,0 +1,12 @@
+{
+  "extends": "_base_phase2.json",
+  "run_name": "p2_1_dcgan",
+  "model": "dcgan",
+  "latent_dim": 100,
+  "ngf": 64,
+  "ndf": 64,
+  "lr_g": 2e-4,
+  "lr_d": 2e-4,
+  "beta1": 0.5,
+  "beta2": 0.999
+}
@@ -0,0 +1,14 @@
+{
+  "extends": "_base_phase2.json",
+  "run_name": "p2_2_wgan",
+  "model": "wgan_basic",
+  "latent_dim": 128,
+  "ngf": 64,
+  "ndf": 64,
+  "lr_g": 1e-4,
+  "lr_d": 1e-4,
+  "beta1": 0.0,
+  "beta2": 0.9,
+  "n_critic": 2,
+  "gp_lambda": 10
+}
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase2.json",
+  "run_name": "p2_3_wgan_sn_attn",
+  "model": "wgan",
+  "image_size": 64,
+  "latent_dim": 128,
+  "ngf": 128,
+  "ndf": 128,
+  "lr_g": 1e-4,
+  "lr_d": 1e-4,
+  "beta1": 0.0,
+  "beta2": 0.9,
+  "n_critic": 2,
+  "gp_lambda": 10
+}
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase2.json",
+  "run_name": "p2_4_wgan_sn_attn_128",
+  "model": "wgan",
+  "image_size": 128,
+  "latent_dim": 128,
+  "ngf": 128,
+  "ndf": 128,
+  "lr_g": 1e-4,
+  "lr_d": 1e-4,
+  "beta1": 0.0,
+  "beta2": 0.9,
+  "n_critic": 2,
+  "gp_lambda": 10
+}