trener
This commit is contained in:
parent
c42a4f6f9a
commit
995b2f6164
3
hft.py
3
hft.py
|
|
@ -171,7 +171,8 @@ training_args = TrainingArguments(
|
||||||
gradient_accumulation_steps=4,
|
gradient_accumulation_steps=4,
|
||||||
learning_rate=2e-5,
|
learning_rate=2e-5,
|
||||||
fp16=True,
|
fp16=True,
|
||||||
logging_steps=100,
|
logging_steps=50, # Częstsze logowanie
|
||||||
|
logging_dir="./logs", # Katalog na logi
|
||||||
save_strategy="steps",
|
save_strategy="steps",
|
||||||
save_steps=1000,
|
save_steps=1000,
|
||||||
report_to="none"
|
report_to="none"
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue