This commit is contained in:
l.gabrysiak 2025-02-26 11:19:33 +01:00
parent f826dc05ca
commit 29c5e40281
1 changed files with 5 additions and 8 deletions

13
gpt.py
View File

@ -90,17 +90,14 @@ def main():
# Konfiguracja treningu
training_args = TrainingArguments(
output_dir="./results",
num_train_epochs=30,
num_train_epochs=32, # Zwiększono liczbę epok
per_device_train_batch_size=2,
learning_rate=1e-5,
weight_decay=0.01,
learning_rate=1e-5, #precyzja uczenia
logging_steps=10,
save_steps=100,
eval_steps=100,
evaluation_strategy="steps",
load_best_model_at_end=True,
weight_decay=0.01,
report_to="none",
save_total_limit=2,
save_strategy="no",
load_best_model_at_end=True, # Ładowanie najlepszego modelu na końcu
)