| accelerate launch --config_file configs/accelerate_config.yaml scripts/train.py \ | |
| --dataset_base_path "/mnt/nfs/workspace/syq/dataset/Hypersim/processed_depth,/mnt/nfs/workspace/syq/dataset/vkitti2" \ | |
| --dataset_metadata_path "./data_split/hypersim_depth/filename_list_train_filtered2.txt,./data_split/vkitti_depth/vkitti_train.txt" \ | |
| --data_file_keys "kontext_images,image" \ | |
| --model_paths "./FLUX.1-Kontext-dev" \ | |
| --learning_rate "1e-5" \ | |
| --num_epochs "8" \ | |
| --remove_prefix_in_ckpt "pipe.dit." \ | |
| --trainable_models "dit" \ | |
| --extra_inputs "kontext_images" \ | |
| --use_gradient_checkpointing \ | |
| --default_caption "Transform to depth map while maintaining original composition" \ | |
| --batch_size "4" \ | |
| --output_path "ckpts/kontext/bs64_sqrt_cons" \ | |
| --eval_file_list "./data_split/nyu_depth/labeled/filename_list_test.txt" \ | |
| --multi_res_noise \ | |
| --save_steps "200" \ | |
| --eval_steps "50" \ | |
| --with_mask \ | |
| --depth_normalization sqrt \ | |
| --dataset_num_workers "16" \ | |
| --extra_loss "cycle_consistency_depth_estimation" \ | |
| --adamw8bit \ | |
| --using_sqrt | |
| # --deterministic_flow | |
| # --extra_loss_start_epoch 0 \ | |
| # --using_sqrt \ | |
| # --resume \ | |