AITextDetector / configs /default.yaml
ChauHPham's picture
Upload folder using huggingface_hub
25faba3 verified
raw
history blame contribute delete
404 Bytes
# Default training/eval configuration
data_path: data/dataset.csv
base_model: roberta-base
save_dir: models/ai_detector
max_length: 256
batch_size: 8
num_epochs: 2
lr: 5e-5
weight_decay: 0.01
logging_steps: 25
eval_strategy: epoch
seed: 42
gradient_accumulation_steps: 1
# Auto-fp16 on CUDA (leave null to auto)
fp16: null
warmup_ratio: 0.0
save_total_limit: 2
save_steps: 0
dataloader_num_workers: 2