{
  "dataloader_num_workers": 0,
  "eval_hf_split": null,
  "eval_split_ratio": 0.0,
  "eval_steps": 100,
  "eval_strategy": "no",
  "gradient_accumulation_steps": 8,
  "gradient_checkpointing": true,
  "hf_dataset": "launch/thinkprm-1K-verification-cots",
  "hf_split": "train",
  "learning_rate": 6e-05,
  "logging_steps": 10,
  "lr_scheduler_type": "constant",
  "max_grad_norm": 1.0,
  "max_length": 4096,
  "model_name_or_path": "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
  "num_train_epochs": 3.0,
  "output_dir": "/mnt/nlp/scratch/home/kopidaki/outputs/thinkprm-full-trl",
  "per_device_eval_batch_size": 1,
  "per_device_train_batch_size": 1,
  "repo_root": "/mnt/nlp/scratch/home/kopidaki/ThinkPRM/",
  "report_to": "wandb",
  "resume_from_checkpoint": null,
  "save_steps": 100,
  "save_strategy": "epoch",
  "save_total_limit": 2,
  "seed": 42,
  "warmup_ratio": 0.0,
  "weight_decay": 0.0
}