Giters

dandelin / ViLT

Code for the ICML 2021 (long talk) paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Debug model

hongzhenwang opened this issue · comments

commented

The training process is DDP, can not debug with pdb. Could you please offer the version code without DDP?
or without pytorch_lighting

commented

Hi @hongzhenwang
Actually, you can debug the code via PDB if you use DDP with a single GPU.
Please try with num_gpus=1.