mod gpt epochs = 5

2025-02-26 09:45:56 +01:00 · 2025-02-26 09:45:56 +01:00 · 8698220dc4
parent 61c45041aa
commit 8698220dc4
1 changed files with 1 additions and 1 deletions
--- a/gpt.py
+++ b/gpt.py
@ -81,7 +81,7 @@ def main():
    # Konfiguracja treningu
    training_args = TrainingArguments(
        output_dir="./results",
-        num_train_epochs=3,  # Zwiększono liczbę epok
+        num_train_epochs=5,  # Zwiększono liczbę epok
        per_device_train_batch_size=2,
        learning_rate=5e-5,
        logging_steps=10,