{
    "version": "0.2.0",
    "configurations": [
        {
            "name": "train mode a (debug)",
            "type": "debugpy",
            "request": "launch",
            "program": "${workspaceFolder}/src/train_cn_sdxl_mc_v2.py",
            "console": "integratedTerminal",
            "justMyCode": false,
            "env": {
                "PYTORCH_ALLOC_CONF": "expandable_segments:True",
                "XFORMERS_DISABLED": "1"
            },
            "args": [
                "--data_root", "/mnt/data1tb/CarlaDS/all_collected_ds",
                "--output_dir", "./checkpoints_debug",
                "--mode", "a",
                "--mixed_precision", "fp16",
                "--image_size", "512",
                "--train_batch_size", "1",
                "--grad_accum", "1",
                "--max_steps", "10",
                "--save_every", "5",
                "--num_workers", "0"
            ]
        },
        {
            "name": "train mode b1 (debug)",
            "type": "debugpy",
            "request": "launch",
            "program": "${workspaceFolder}/src/train_cn_sdxl_mc_v2.py",
            "console": "integratedTerminal",
            "justMyCode": false,
            "env": {
                "PYTORCH_ALLOC_CONF": "expandable_segments:True",
                "XFORMERS_DISABLED": "1"
            },
            "args": [
                "--data_root", "/mnt/data1tb/CarlaDS/all_collected_ds",
                "--output_dir", "./checkpoints_debug",
                "--mode", "b1",
                "--mixed_precision", "fp16",
                "--image_size", "512",
                "--train_batch_size", "1",
                "--grad_accum", "1",
                "--max_steps", "5",
                "--save_every", "5",
                "--num_workers", "0"
            ]
        },
        {
            "name": "train full (mode a)",
            "type": "debugpy",
            "request": "launch",
            "program": "${workspaceFolder}/src/train_cn_sdxl_mc_v2.py",
            "console": "integratedTerminal",
            "justMyCode": false,
            "env": {
                "PYTORCH_ALLOC_CONF": "expandable_segments:True",
                "XFORMERS_DISABLED": "1"
            },
            "args": [
                "--data_root", "/mnt/data1tb/CarlaDS/all_collected_ds",
                "--output_dir", "./checkpoints",
                "--mode", "a",
                "--mixed_precision", "fp16",
                "--image_size", "512",
                "--train_batch_size", "1",
                "--grad_accum", "8",
                "--max_steps", "50000",
                "--save_every", "2000",
                "--num_workers", "4"
            ]
        }
    ]
}