Add model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- reinforcement-learning
|
| 5 |
+
- ddpg
|
| 6 |
+
- diffusion
|
| 7 |
+
- gpt
|
| 8 |
+
- pytorch
|
| 9 |
+
- painting
|
| 10 |
+
library_name: pytorch
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# Mirabilia — Checkpoints
|
| 14 |
+
|
| 15 |
+
Poids entraînés du projet **[Mirabilia_Painter](https://github.com/RaphGonz/Mirabilia_Painter)**,
|
| 16 |
+
une réimplémentation de *"Learning to Paint"* (DDPG + renderer neuronal différentiable) et
|
| 17 |
+
ses évolutions, documentées publiquement dans la série **Mirabilia**.
|
| 18 |
+
|
| 19 |
+
Un agent RL apprend à peindre une image cible en posant des traits rectangulaires opaques
|
| 20 |
+
sur une toile 64×64, guidé par un signal L2 incrémental.
|
| 21 |
+
|
| 22 |
+
## Contenu du repo
|
| 23 |
+
|
| 24 |
+
| Dossier | Modèle | Description |
|
| 25 |
+
|---------|--------|-------------|
|
| 26 |
+
| `painter/` | DDPG (le peintre) | `actor.pt`, `critic.pt`, `disc.pt`, `disc_target.pt`, `opt.pt` |
|
| 27 |
+
| `llm/` | GPT decoder-only (from scratch) | `gpt.pt` (poids), `opt.pt` (optimiseur) par itération |
|
| 28 |
+
| `diffusion/` | UNet (diffusion) | `unet.pt`, `unet_ema.pt`, `opt.pt` par époque |
|
| 29 |
+
| `diffusion_cache/` | Embeddings | Embeddings texte précalculés (reproductibilité) |
|
| 30 |
+
|
| 31 |
+
Chaque sous-dossier contient un checkpoint par étape d'entraînement
|
| 32 |
+
(ex. `painter/ep_006000/`, `llm/iter_720000/`, `diffusion/ep_002000/`).
|
| 33 |
+
|
| 34 |
+
Les fichiers `opt.pt` (états de l'optimiseur) ne sont utiles que pour **reprendre**
|
| 35 |
+
un entraînement ; pour utiliser un modèle, seuls les poids (`actor.pt`, `gpt.pt`,
|
| 36 |
+
`unet_ema.pt`) sont nécessaires.
|
| 37 |
+
|
| 38 |
+
## Utilisation
|
| 39 |
+
|
| 40 |
+
```python
|
| 41 |
+
from huggingface_hub import hf_hub_download
|
| 42 |
+
import torch
|
| 43 |
+
|
| 44 |
+
# Exemple : charger le peintre DDPG final
|
| 45 |
+
path = hf_hub_download(repo_id="RaphGonz/Mirabilia", filename="painter/ep_006000/actor.pt")
|
| 46 |
+
state = torch.load(path, map_location="cpu")
|
| 47 |
+
```
|
| 48 |
+
|
| 49 |
+
Voir le dépôt GitHub pour l'architecture complète, l'environnement et les scripts d'entraînement.
|
| 50 |
+
|
| 51 |
+
## Stack
|
| 52 |
+
|
| 53 |
+
PyTorch 2.7 · CUDA 12.x · entraînement local (GPU NVIDIA).
|