CUDA_VISIBLE_DEVICES="0,1,2,3" accelerate launch train.py \
--output_dir "output_save_dir" \
--wandb_name "wandb_name" \
--wandb_project "wandb_project" \
--log_freq 100 \
--checkpointing_steps 5000 \
--validation_steps 5000 \
--max_train_steps 100000 \
--learning_rate 3e-4 \
--train_batch_size 2 \
--gradient_accumulation_steps 1 \
--mixed_precision 'no' \
--report_to "tensorboard" \
--seed 123 \
--train_dataset_txt_paths_list_lq '/dataset_path/sidd_dataset_input.txt','/dataset_path/gopro_dataset_lq.txt' \
--train_dataset_txt_paths_list_gt '/dataset_path/sidd_dataset_gt.txt','/dataset_path/gopro_dataset_gt.txt' \
--train_dataset_txt_paths_list_gt_refined_2x '/dataset_path/sidd_dataset_gt_2x.txt','/dataset_path/gopro_dataset_gt_2x.txt' \
--train_dataset_txt_paths_list_gt_refined_3x '/dataset_path/sidd_dataset_gt_3x.txt','/dataset_path/gopro_dataset_gt_3x.txt' \
--train_dataset_txt_paths_list_gt_refined_4x '/dataset_path/sidd_dataset_gt_4x.txt','/dataset_path/gopro_dataset_gt_4x.txt' \
--test_dataset_txt_paths_list_lq '/dataset_path/gopro_test_lq.txt' \
--test_dataset_txt_paths_list_gt '/dataset_path/gopro_test_gt.txt' \
--dataset_prob_paths_list 1,15 \
--lambda_l2 1 \
--tracker_project_name "refinement" \
--pretrained_mask_generator_path "path/" \
