mod gpt epochs = 5
This commit is contained in:
parent
61c45041aa
commit
8698220dc4
2
gpt.py
2
gpt.py
|
|
@ -81,7 +81,7 @@ def main():
|
||||||
# Konfiguracja treningu
|
# Konfiguracja treningu
|
||||||
training_args = TrainingArguments(
|
training_args = TrainingArguments(
|
||||||
output_dir="./results",
|
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,
|
per_device_train_batch_size=2,
|
||||||
learning_rate=5e-5,
|
learning_rate=5e-5,
|
||||||
logging_steps=10,
|
logging_steps=10,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue