kto_lora_argument.json

{
    "model_name_or_path": "IEITYuan/Yuan2-2B",
    "train_dataset_path": "./data/train.jsonl",
    "dev_dataset_path": "./data/dev.jsonl",
    "output_dir": "./checkpoints/kto_ckpts",
    "per_device_train_batch_size": 1,
    "gradient_accumulation_steps": 8,
    "per_device_eval_batch_size": 1,
    "num_train_epochs": 1,
    "max_steps": 100,
    "learning_rate": 2e-05,
    "warmup_steps": 10,
    "logging_steps": 1,
    "evaluation_strategy": "steps",
    "save_strategy": "steps",
    "eval_steps": 100,
    "save_steps": 500,
    "max_seq_len": 4096,
    "max_prompt_len": 2048,
    "bf16": true,
    "fp16_opt_level": "O2",
    "do_train": true,
    "do_eval": true,
    "disable_tqdm": true,
    "load_best_model_at_end": true,
    "tensor_parallel_degree": 8,
    "sharding": "stage1",
    "use_flash_attention": true,
    "recompute": false,
    "recompute_granularity": "full",
    "beta": 0.1,
    "benchmark": false,
    "unified_checkpoint": true,
    "autotuner_benchmark":false,
    "lazy": false,
    "seed":42,
    "lora": true
  }