Downloads · 30 days
0
pokemon-gan
andreribeiro87/pokemon-gan
pokemon-gan is a machine learning model from andreribeiro87. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch.
Access
Public
Updated Mar 25, 2026
Repo size
3.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth3.7 MB · 100%
At a glance
- Library
- pytorch
- Access
- Public
- Created
- Mar 25, 2026
- Updated
- Mar 25, 2026
- SHA
- 11b2ade8
Model card
From the Hugging Face model README
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"
}
- Library
- pytorch
- Languages
- en
- Created
- Mar 25, 2026
- Updated
- Mar 25, 2026