training_args.json

{
  "lora_r": 16,
  "lora_alpha": 16,
  "lora_dropout": 0.05,
  "lora_target_modules": "q_proj,v_proj",
  "tuning_strategy": "lora",
  "num_trainable_layers": 5,
  "output_dir": "checkpoints/pure/Chest_xray_pa/medgemma_lora",
  "num_train_epochs": 3,
  "learning_rate": 0.0002,
  "per_device_train_batch_size": 16,
  "per_device_eval_batch_size": 16,
  "gradient_accumulation_steps": 2,
  "logging_steps": 10,
  "eval_steps": 20,
  "save_steps": 20,
  "warmup_ratio": 0.03,
  "weight_decay": 0.0,
  "max_grad_norm": 0.3,
  "lr_scheduler_type": "constant",
  "save_strategy": "steps",
  "bf16": true,
  "tf32": false,
  "gradient_checkpointing": true,
  "optim": "adamw_torch_fused",
  "ft_type": "SFT",
  "data_name": "Chest_xray_pa",
  "sft_type": "pure",
  "balanced": false
}