23 lines
852 B
JSON
23 lines
852 B
JSON
{
|
|
"base_model": "/NHNHOME/26msit001_A/BASE/edge_ai_lab/jongbokwon/warp-safety/outputs/chat_ssft/llama2_7b_chat_circuit_breakers",
|
|
"fine_tuning_type": "Full Parameter Fine-tuning",
|
|
"dataset": "boolq",
|
|
"dataset_path": "/NHNHOME/26msit001_A/BASE/edge_ai_lab/jongbokwon/Safety-WaRP-LLM/data/commonsense/boolq_train.json",
|
|
"num_train_samples": 0,
|
|
"batch_size": 4,
|
|
"grad_accum": 4,
|
|
"learning_rate": 5e-05,
|
|
"weight_decay": 0.01,
|
|
"warmup_ratio": 0.1,
|
|
"epochs": 3,
|
|
"max_length": 1024,
|
|
"max_grad_norm": 1.0,
|
|
"lr_scheduler_type": "cosine",
|
|
"optimizer": "AdamW (torch)",
|
|
"gradient_checkpointing": false,
|
|
"dtype": "bf16",
|
|
"trainer_type": "Trainer",
|
|
"safety_mix_ratio": 0.1,
|
|
"safety_data_path": "/NHNHOME/26msit001_A/BASE/edge_ai_lab/jongbokwon/Safety-WaRP-LLM/data/circuit_breakers_train.json",
|
|
"geometry_trace_dir": null
|
|
} |