77 lines
2.7 KiB
JSON
77 lines
2.7 KiB
JSON
{
|
|
"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"
|
|
]
|
|
}
|
|
]
|
|
} |