File tree Expand file tree Collapse file tree 1 file changed +29
-0
lines changed
Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Original file line number Diff line number Diff line change 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+ }
You can’t perform that action at this time.
0 commit comments