error when run demo.py - Githubissues

VDIGPKU / M2Det

M2Det: A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network

MIT License

1.45k stars 318 forks source link

Open kaishijeng opened 4 years ago

kaishijeng commented 4 years ago

I have encountered an error when I run: python demo.py -c=configs/m2det512_vgg.py -m=weights/m2det512_vgg.pth --show

Error: RuntimeError: max_pool2d_with_indices_out_cuda_frame failed with error code 0

Any idea why this happens?

Thanks