RaphGonz commited on
Commit
ea1b939
·
verified ·
1 Parent(s): 6682c08

Add model card

Browse files
Files changed (1) hide show
  1. README.md +53 -0
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).