bogdanpavl/SDXL_ControlNet_StreetView_Gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
pikaliov
2026-04-16 15:35:01 +03:00
commit 30846d68ad
25 changed files with 2643 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
.vscode/launch.json vendored Normal file
View File

@@ -0,0 +1,77 @@
{
"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"
]
}
]
}