Phase 5 preparation

generator/configs/phase5/p5_ddpm.json

@@ -1,7 +1,7 @@
 {
   "run_name": "p5_ddpm",
   "model": "ddpm",
-  "epochs": 200,
+  "epochs": 400,
   "data_dir": "cropped/generator",
   "sources": ["wiki"],
   "augment": "hflip",
@@ -18,5 +18,7 @@
   "ddim_steps": 100,
   "sample_interval": 10,
   "fid_interval": 25,
-  "fid_n_real": 5000
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
 }

generator/configs/phase5/p5_gan.json

@@ -1,11 +1,11 @@
 {
   "run_name": "p5_gan",
   "model": "wgan",
-  "epochs": 200,
+  "epochs": 400,
   "data_dir": "cropped/generator",
   "sources": ["wiki"],
   "augment": true,
-  "image_size": 128,
+  "image_size": 64,
   "latent_dim": 128,
   "ngf": 128,
   "ndf": 128,
@@ -17,5 +17,7 @@
   "gp_lambda": 10,
   "sample_interval": 10,
   "fid_interval": 25,
-  "fid_n_real": 5000
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
 }

generator/configs/phase5/p5_vae.json

@@ -1,7 +1,7 @@
 {
   "run_name": "p5_vae",
   "model": "vae",
-  "epochs": 200,
+  "epochs": 400,
   "data_dir": "cropped/generator",
   "sources": ["wiki"],
   "augment": "hflip",
@@ -10,11 +10,13 @@
   "ngf": 64,
   "lr": 1e-3,
   "lr_d": 1e-4,
-  "beta_kl": 0.0001,
+  "beta_kl": 0.25,
   "lambda_perceptual": 0.1,
   "lambda_adversarial": 0.1,
   "ndf_patch": 64,
   "sample_interval": 10,
   "fid_interval": 25,
-  "fid_n_real": 5000
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
 }