For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

PatrickHua / SimSiam

A pytorch implementation for paper 'Exploring Simple Siamese Representation Learning'

MIT License

814 stars 135 forks source link

For debugging consider passing CUDA_LAUNCH_BLOCKING=1. #53

Open fhaji75 opened 1 year ago

fhaji75 commented 1 year ago

I want to train on a custom dataset, but I got multiple errors I tried to solve that, one error is For debugging, consider passing CUDA_LAUNCH_BLOCKING=1. I couldn't solve this, I ran the repository in Colab.