Upload folder using huggingface_hub

best_checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:826a00120a0510635d90be877ed76fa23270bf19bb81fec6dcc6294c65e31288
+size 403768367

readMe.md

@@ -0,0 +1,70 @@
+---
+tags:
+- MuseMachine
+- pytorch
+- generative-adversarial-network
+- variational-autoencoder
+---
+
+# MuseMachine
+
+A modular AE-GAN pipeline for portrait generation.
+
+## Architecture
+
+| Component | Description |
+|---|---|
+| Encoder | VAE encoder producing latent representations |
+| Generator | Decodes latent vectors into coarse images |
+| Refiner | Iteratively refines generations over 4 steps |
+| Discriminator | PatchGAN-style discriminator |
+
+## Hyperparameters
+
+```json
+{
+  "image_size": 256,
+  "latent_dim": 256,
+  "batch_size": 4,
+  "epochs": 100,
+  "lr_g": 0.0002,
+  "lr_d": 5e-05,
+  "alpha_recon": 100.0,
+  "beta_adv": 1.0,
+  "beta_kl": 0.05,
+  "noise_std": 0.15,
+  "refinement_steps": 4,
+  "refiner_step_size": 0.08,
+  "refine_decay": 0.8,
+  "dataset_name": "ajehsmihba/aesthetic-female-portraits",
+  "outputs_dir": "outputs",
+  "models_dir": "models",
+  "save_every": 50,
+  "push_to_hf": true,
+  "hf_repo_id": "ajehsmihba/MuseMachine",
+  "grad_clip": 1.0,
+  "vis_every": 10
+}
+```
+
+## Best Metrics
+
+```json
+{
+  "D": 1.0980566523291848,
+  "EG": 88.96168032559481,
+  "KL": 337.3994445800781,
+  "Recon": 0.1864572438326749,
+  "Coarse_Recon": 0.19136723740534348
+}
+```
+
+## Resume Training
+
+```python
+from huggingface_hub import hf_hub_download
+import torch
+
+ckpt_path = hf_hub_download(repo_id="ajehsmihba/MuseMachine", filename="best_checkpoint.pth")
+ckpt = torch.load(ckpt_path, map_location="cpu")
+```