From f826dc05ca7b431c4d417bf962a55998787a50c2 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Wed, 26 Feb 2025 11:17:48 +0100 Subject: [PATCH] mod gpt --- gpt.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/gpt.py b/gpt.py index 7fff1ac..7eb52e3 100644 --- a/gpt.py +++ b/gpt.py @@ -89,19 +89,19 @@ def main(): # Konfiguracja treningu training_args = TrainingArguments( - output_dir="./results", - num_train_epochs=30, - per_device_train_batch_size=2, - learning_rate=1e-5, - weight_decay=0.01, - logging_steps=10, - save_steps=100, - eval_steps=100, - evaluation_strategy="steps", - load_best_model_at_end=True, - report_to="none", - save_total_limit=2, -) + output_dir="./results", + num_train_epochs=30, + per_device_train_batch_size=2, + learning_rate=1e-5, + weight_decay=0.01, + logging_steps=10, + save_steps=100, + eval_steps=100, + evaluation_strategy="steps", + load_best_model_at_end=True, + report_to="none", + save_total_limit=2, + ) # Trainer