keithkctse

Cannot specify GPU when carrying out inference with mask DINO

4

When I set cfg.MODEL.DEVICE = 'cuda:1' ,it says illegal memory access. The problem does not exist when i set cfg.MODEL.DEVICE = 'cuda:0', how to solve this? 

Model Weights on Google Drive

7

Could you please upload the pre-trained weights of ISTR-SMT to Google Drive too? I found it difficult to download from Baidu, thanks a lot!