Skip to content

Commit 1bc2a3c

Browse files
committed
VSCode launch file for example execution.
1 parent db1e28b commit 1bc2a3c

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

.vscode/launch.json

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
{
2+
// Use IntelliSense to learn about possible attributes.
3+
// Hover to view descriptions of existing attributes.
4+
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
5+
"version": "0.2.0",
6+
"configurations": [
7+
{
8+
"name": "Python: ppi_finetuning",
9+
"type": "python",
10+
"request": "launch",
11+
"program": "${workspaceFolder}/src/ppi_finetuning.py",
12+
"console": "integratedTerminal",
13+
"justMyCode": false,
14+
"args": [
15+
// "--help"
16+
"--accelerator", "gpu",
17+
"--devices", "1",
18+
// "--devices", "2", // for 2 gpus
19+
"--num_sanity_val_steps", "0",
20+
// "--limit_train_batches", "2",
21+
// "--limit_val_batches", "2",
22+
],
23+
"env": {
24+
"CUDA_VISIBLE_DEVICES": "0"
25+
// "CUDA_VISIBLE_DEVICES": "0,1" // for 2 gpus
26+
}
27+
}
28+
]
29+
}

0 commit comments

Comments
 (0)