pokemon-gan

Local run name: pokemon_trial15_sharpen

Generator weights exported from the Week 6 GAN lab training codebase.

Repository files

Key	Value
pokemon_gan_generator.pth	Generator state_dict (PyTorch)
generator_training_config.json	Training hyperparameters / layout (JSON)
README.md	This model card (auto-generated on upload)

Export provenance

Key	Value
Checkpoint path	/home/ucloud/caa-andre/complements-ml-labs-andreribeiro87/week-06/outputs/runs/pokemon_trial15_sharpen__20260323T021012Z__51f37f23/checkpoints/latest.pt
Run directory JSON	config.json, run_meta.json

Run metadata (run_meta.json)

Key	Value
config_fingerprint	51f37f23
created_utc	20260323T021012Z
run_name	pokemon_trial15_sharpen

Training configuration (summary)

Key	Value
run_name	pokemon_trial15_sharpen
task	pokemon
image_size	32
nz	64
ngf	32
ndf	32
num_classes	10
adversarial_loss	wgan
use_gradient_penalty	True
lambda_fm	0.4
lambda_perc	0.1
epochs	500
max_steps	2000000
batch_size	32
lr_g	0.00015
lr_d	0.0004926710526703187
n_critic	2
seed	42
spectral_norm_d	False

Source JSON (run directory)

Verbatim JSON files from the run folder root (next to checkpoints/).

config.json

{
  "adversarial_loss": "wgan",
  "batch_size": 32,
  "beta1": 0.32601223625614445,
  "beta2": 0.9940362161905395,
  "checkpoint_every": 500,
  "data_cache_dir": null,
  "device": "cuda",
  "epochs": 500,
  "eval_at_end": true,
  "eval_batch_size": 32,
  "eval_every": 500,
  "eval_num_fake": 2048,
  "eval_num_real": 2048,
  "image_size": 32,
  "lambda_fm": 0.4,
  "lambda_gp": 19.21339982391642,
  "lambda_perc": 0.1,
  "log_every": 50,
  "lr_d": 0.0004926710526703187,
  "lr_g": 0.00015,
  "max_steps": 2000000,
  "metrics_device": null,
  "n_critic": 2,
  "n_samples_grid": 64,
  "ndf": 32,
  "ngf": 32,
  "num_classes": 10,
  "num_workers": 2,
  "nz": 64,
  "output_root": "outputs",
  "run_name": "pokemon_trial15_sharpen",
  "sample_every": 500,
  "seed": 42,
  "spectral_norm_d": false,
  "tags": {
    "continued_run_dir": "pokemon_trial15_long__20260322T183621Z__fc4c266d",
    "hpo": "1",
    "optuna_study": "gan_hpo",
    "optuna_trial": "15",
    "resumed_ckpt": "latest.pt"
  },
  "task": "pokemon",
  "use_bf16": true,
  "use_gradient_penalty": true,
  "use_wandb": true,
  "wandb_entity": null,
  "wandb_mode": "online",
  "wandb_project": "week06-gan"
}

run_meta.json

{
  "config": {
    "adversarial_loss": "wgan",
    "batch_size": 32,
    "beta1": 0.32601223625614445,
    "beta2": 0.9940362161905395,
    "checkpoint_every": 500,
    "data_cache_dir": null,
    "device": "cuda",
    "epochs": 500,
    "eval_at_end": true,
    "eval_batch_size": 32,
    "eval_every": 500,
    "eval_num_fake": 2048,
    "eval_num_real": 2048,
    "image_size": 32,
    "lambda_fm": 0.4,
    "lambda_gp": 19.21339982391642,
    "lambda_perc": 0.1,
    "log_every": 50,
    "lr_d": 0.0004926710526703187,
    "lr_g": 0.00015,
    "max_steps": 2000000,
    "metrics_device": null,
    "n_critic": 2,
    "n_samples_grid": 64,
    "ndf": 32,
    "ngf": 32,
    "num_classes": 10,
    "num_workers": 2,
    "nz": 64,
    "output_root": "outputs",
    "run_name": "pokemon_trial15_sharpen",
    "sample_every": 500,
    "seed": 42,
    "spectral_norm_d": false,
    "tags": {
      "continued_run_dir": "pokemon_trial15_long__20260322T183621Z__fc4c266d",
      "hpo": "1",
      "optuna_study": "gan_hpo",
      "optuna_trial": "15",
      "resumed_ckpt": "latest.pt"
    },
    "task": "pokemon",
    "use_bf16": true,
    "use_gradient_penalty": true,
    "use_wandb": true,
    "wandb_entity": null,
    "wandb_mode": "online",
    "wandb_project": "week06-gan"
  },
  "config_fingerprint": "51f37f23",
  "created_utc": "20260323T021012Z",
  "run_name": "pokemon_trial15_sharpen"
}