FineTurnX/test_training_args.py

"""
测试 TrainingArguments 参数
"""

from transformers import TrainingArguments
import torch

print("测试 TrainingArguments 参数...")

try:
    args = TrainingArguments(
        output_dir="./test_output",
        num_train_epochs=3,
        per_device_train_batch_size=1,
        gradient_accumulation_steps=4,
        learning_rate=2e-4,
        warmup_ratio=0.03,
        weight_decay=0.01,
        logging_steps=10,
        save_steps=50,
        eval_strategy="no",
        save_total_limit=3,
        fp16=torch.cuda.is_available(),
        optim="paged_adamw_32bit",
        lr_scheduler_type="cosine",
        report_to="none",
        remove_unused_columns=False,
    )
    print("✓ TrainingArguments 参数验证通过！")
    print(f"  输出目录：{args.output_dir}")
    print(f"  训练轮数：{args.num_train_epochs}")
    print(f"  FP16: {args.fp16}")
except Exception as e:
    print(f"✗ TrainingArguments 参数验证失败：{e}")
    import traceback
    traceback.print_exc()