thijn/sa_ai_training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
thijn 2025-04-01 11:06:12 +02:00
parent e3e5deb32f
commit 467671ffca
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -41,6 +41,7 @@ training_args = TrainingArguments(
warmup_steps=500, warmup_steps=500,
weight_decay=0.01, weight_decay=0.01,
logging_dir='./logs', logging_dir='./logs',
save_total_limit=5,
fp16=True # Enable Mixed Precision Training for GPU fp16=True # Enable Mixed Precision Training for GPU
) )