Skip to content

Commit

Permalink
Added NanoGpt and update website
Browse files Browse the repository at this point in the history
  • Loading branch information
KenN7 committed Mar 27, 2024
1 parent 28af9ae commit e8f1cb2
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions docs/docs/nanogpt.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,8 @@ Pour lancer un entrainement avec le corpus de Shakespeare il suffit de lancer, p
poetry run python main.py --train --save_model my_new_model.pth --block_size 512 --eval_iters 10000
```

Il est aussi possible de changer le corpus d'entrainement avec le paramètre `--input` (par défaut celui-ci le programme prend le fichier `input.txt` contenant le corpus de Shakespeare).

Par la suite, une fois entrainé, pour tester votre modèle:

```bash
Expand All @@ -54,6 +56,8 @@ Toute une liste de paramètres supplémentaires sont configurables :

* `--infer` : Mode inférence

* `--input` : fichier d'entrée de corpus d'entrainement

* `--save_model SAVE_MODEL` : Sauvegarde le modèle dans le fichier spécifié

* `--load_model LOAD_MODEL` : Charge le modèle depuis le fichier spécifié
Expand Down

0 comments on commit e8f1cb2

Please sign in to comment.