52
Browse files- app/app.py +1 -1
app/app.py
CHANGED
|
@@ -9,7 +9,7 @@ def prepare_data():
|
|
| 9 |
|
| 10 |
def train_model():
|
| 11 |
st.text("Entraînement du modèle en cours...")
|
| 12 |
-
subprocess.run(["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):
|
|
|
|
| 9 |
|
| 10 |
def train_model():
|
| 11 |
st.text("Entraînement du modèle en cours...")
|
| 12 |
+
subprocess.run(["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):
|