Notebooks Classificador

2026-05-14 16:20:33 +01:00
parent 9ae334410d
commit 2062a91985
734 changed files with 75472 additions and 1730 deletions
@@ -0,0 +1,7 @@
+{
+  "data_dir": "data",
+  "sources": ["wiki"],
+  "augment": false,
+  "image_size": 64,
+  "batch_size": 64
+}
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_ddpm",
+  "model": "ddpm",
+  "base_ch": 128,
+  "T": 1000,
+  "beta_start": 0.0001,
+  "beta_end": 0.02,
+  "dropout": 0.1,
+  "epochs": 200,
+  "lr": 0.0001,
+  "subsample": 1.0,
+  "sample_interval": 10,
+  "sample_steps": 100
+}
@@ -0,0 +1,15 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_ddpm_small",
+  "model": "ddpm",
+  "base_ch": 64,
+  "T": 1000,
+  "beta_start": 0.0001,
+  "beta_end": 0.02,
+  "dropout": 0.1,
+  "epochs": 100,
+  "lr": 0.0002,
+  "subsample": 0.2,
+  "sample_interval": 10,
+  "sample_steps": 100
+}
@@ -0,0 +1,12 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_vae",
+  "model": "vae",
+  "latent_dim": 256,
+  "epochs": 100,
+  "lr": 0.0005,
+  "beta": 1.0,
+  "perceptual_weight": 0.1,
+  "subsample": 1.0,
+  "sample_interval": 5
+}
@@ -0,0 +1,17 @@
+{
+  "extends": "_base_phase0.json",
+  "run_name": "p0_wgan",
+  "model": "wgan",
+  "latent_dim": 128,
+  "ngf": 64,
+  "ndf": 64,
+  "epochs": 200,
+  "lr_g": 0.0001,
+  "lr_d": 0.0001,
+  "beta1": 0.0,
+  "beta2": 0.9,
+  "n_critic": 5,
+  "gp_lambda": 10,
+  "subsample": 1.0,
+  "sample_interval": 10
+}
@@ -0,0 +1,13 @@
+{
+  "model": "dcgan",
+  "image_size": 64,
+  "latent_dim": 100,
+  "ngf": 64,
+  "ndf": 64,
+  "epochs": 50,
+  "lr_g": 2e-4,
+  "lr_d": 2e-4,
+  "beta1": 0.5,
+  "beta2": 0.999,
+  "augment": false
+}
@@ -0,0 +1,5 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1a_dcgan_128",
+  "image_size": 128
+}
@@ -0,0 +1,5 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1a_dcgan_64",
+  "image_size": 64
+}
@@ -0,0 +1,5 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1b_dcgan_aligned",
+  "data_dir": "cropped/generator"
+}
@@ -0,0 +1,5 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1b_dcgan_full",
+  "data_dir": "data"
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1c_dcgan_full_aug",
+  "data_dir": "cropped/generator",
+  "augment": true
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1c_dcgan_hflip",
+  "data_dir": "cropped/generator",
+  "augment": false
+}
@@ -0,0 +1,5 @@
+{
+  "extends": "_base_dcgan.json",
+  "run_name": "p1d_dcgan_combined",
+  "data_dir": ["data", "cropped/generator"]
+}
@@ -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
+}
@@ -0,0 +1,14 @@
+{
+  "epochs": 100,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": "hflip",
+  "image_size": 64,
+  "model": "vae",
+  "latent_dim": 256,
+  "ngf": 64,
+  "grad_clip": 1.0,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_1_vae",
+  "lr": 5e-4,
+  "beta_kl": 0.25,
+  "lambda_perceptual": 0.0,
+  "lambda_adversarial": 0.0
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_2_vae_perceptual",
+  "lr": 5e-4,
+  "beta_kl": 0.25,
+  "lambda_perceptual": 0.1,
+  "lambda_adversarial": 0.0
+}
@@ -0,0 +1,10 @@
+{
+  "extends": "_base_phase3.json",
+  "run_name": "p3_3_vae_patchgan",
+  "lr": 5e-4,
+  "lr_d": 1e-4,
+  "beta_kl": 0.25,
+  "lambda_perceptual": 0.1,
+  "lambda_adversarial": 0.01,
+  "ndf_patch": 64
+}
@@ -0,0 +1,19 @@
+{
+  "epochs": 100,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": "hflip",
+  "image_size": 64,
+  "model": "ddpm",
+  "T": 1000,
+  "ddim_steps": 100,
+  "lr": 2e-4,
+  "base_ch": 128,
+  "ch_mult": [1, 2, 2, 2],
+  "attn_resolutions": [16, 8],
+  "num_res_blocks": 2,
+  "dropout": 0.1,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_1_ddpm_linear",
+  "noise_schedule": "linear",
+  "pred_type": "eps"
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_2_ddpm_cosine",
+  "noise_schedule": "cosine",
+  "pred_type": "eps"
+}
@@ -0,0 +1,6 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_3_ddpm_vpred",
+  "noise_schedule": "cosine",
+  "pred_type": "v"
+}
@@ -0,0 +1,8 @@
+{
+  "extends": "_base_phase4.json",
+  "run_name": "p4_4_ddpm_wider",
+  "noise_schedule": "cosine",
+  "pred_type": "v",
+  "base_ch": 192,
+  "attn_resolutions": [32, 16, 8]
+}
@@ -0,0 +1,24 @@
+{
+  "run_name": "p5_ddpm",
+  "model": "ddpm",
+  "epochs": 1000,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": "hflip",
+  "image_size": 64,
+  "T": 1000,
+  "noise_schedule": "cosine",
+  "pred_type": "v",
+  "base_ch": 192,
+  "ch_mult": [1, 2, 2, 2],
+  "attn_resolutions": [32, 16, 8],
+  "num_res_blocks": 2,
+  "dropout": 0.1,
+  "lr": 2e-4,
+  "ddim_steps": 100,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
+}
@@ -0,0 +1,23 @@
+{
+  "run_name": "p5_gan",
+  "model": "wgan",
+  "epochs": 1000,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": true,
+  "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,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
+}
@@ -0,0 +1,22 @@
+{
+  "run_name": "p5_vae",
+  "model": "vae",
+  "epochs": 1000,
+  "data_dir": "cropped/generator",
+  "sources": ["wiki"],
+  "augment": "hflip",
+  "image_size": 64,
+  "latent_dim": 256,
+  "ngf": 64,
+  "lr": 1e-3,
+  "lr_d": 1e-4,
+  "beta_kl": 0.25,
+  "lambda_perceptual": 0.1,
+  "lambda_adversarial": 0.01,
+  "ndf_patch": 64,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000,
+  "max_train_hours": 24,
+  "fid_patience": 2
+}
@@ -0,0 +1,11 @@
+{
+  "batch_size": 64,
+  "ema_decay": 0.9999,
+  "data_dir": "data",
+  "sources": ["wiki"],
+  "subsample": 1.0,
+  "sample_interval": 10,
+  "fid_interval": 25,
+  "fid_n_real": 5000,
+  "num_workers": 2
+}