How can I debug training part code?

[TwiceMao]

@jytime thanks for your awesome work! I want to finetune your checkpoint based on my own dataset. And the codes are needed to be changed. So I want to know how can I debug the code in vscode? How should I configure the launch.json of Vscode?

4. Fine-tuning on Co3DTo fine-tune the provided pre-trained model on the Co3D dataset, run the following command. This example uses 4 GPUs with PyTorch Distributed Data Parallel (DDP):torchrun --nproc_per_node=4 launch.py

[jytime]

Hi if I remember correclty, you just need to set num_workers=0, and then you can set breakpoints in vscode, or use pdb for debugging