args.json

{
  "pretrained_model": "./experiments/baseline/checkpoint/latest.pth",
  "args_file": "./experiments/baseline/checkpoint/args.json",
  "train_set": "./preprocessed_data",
  "train_set_list": "/home/Humble/Humble/EndtoEnd/YSH-Diffusion-Planner/diffusion_planner_training.json",
  "num_samples": 8,
  "beta": 1.0,
  "weight_clip": 10.0,
  "temperature": 1.0,
  "learning_rate": 1e-05,
  "gradient_accumulation_steps": 1,
  "reward_type": "base",
  "epochs": 50,
  "batch_size": 256,
  "num_workers": 16,
  "save_freq": 5,
  "output_dir": "./wr_output",
  "name": "wr-base",
  "use_mlflow": true,
  "mlflow_tracking_uri": "https://dagshub.com/YSH-research/YSH-Diffusion-Planner.mlflow",
  "mlflow_experiment_name": "wr-finetuning",
  "local_rank": -1
}