53
Browse files- app/app.py +2 -2
app/app.py
CHANGED
|
@@ -4,12 +4,12 @@ import subprocess
|
|
| 4 |
|
| 5 |
def prepare_data():
|
| 6 |
st.text("Préparation des données en cours...")
|
| 7 |
-
subprocess.
|
| 8 |
st.text("Préparation des données terminée.")
|
| 9 |
|
| 10 |
def train_model():
|
| 11 |
st.text("Entraînement du modèle en cours...")
|
| 12 |
-
subprocess.
|
| 13 |
st.text("Entraînement du modèle terminé.")
|
| 14 |
|
| 15 |
def run_command(command):
|
|
|
|
| 4 |
|
| 5 |
def prepare_data():
|
| 6 |
st.text("Préparation des données en cours...")
|
| 7 |
+
subprocess.run_command(["python3", "../data/shakespeare_char/prepare.py"])
|
| 8 |
st.text("Préparation des données terminée.")
|
| 9 |
|
| 10 |
def train_model():
|
| 11 |
st.text("Entraînement du modèle en cours...")
|
| 12 |
+
subprocess.run_command(["python3", "../train.py", "../config/train_shakespeare_char.py", "--device=cpu", "--compile=False", "--eval_iters=20", "--log_interval=1", "--block_size=64", "--batch_size=12", "--n_layer=4", "--n_head=4", "--n_embd=128", "--max_iters=2000", "--lr_decay_iters=2000", "--dropout=0.0"])
|
| 13 |
st.text("Entraînement du modèle terminé.")
|
| 14 |
|
| 15 |
def run_command(command):
|