Views
No views yet
1training_args = TrainingArguments(
2 output_dir="./gemma_fine_tuned",
3 per_device_train_batch_size=2,
4 gradient_accumulation_steps=8,
5 learning_rate=1e-4,
6 num_train_epochs=3,
7 weight_decay=0.01,
8 warmup_steps=50,
9 save_steps=100,
10 logging_steps=20,
11 fp16=False,
12 bf16=True,
13 save_total_limit=2,
14 report_to='none',
15 gradient_checkpointing=True
16)