RuntimeError Traceback (most recent call last)
in
2
3 if cuda_flag:
----> 4 torch.cuda.empty_cache()
5
6 batch_size = 40

~\AppData\Local\Continuum\anaconda3\lib\site-packages\torch\cuda_init_.py in empty_cache()
424 """
425 if _initialized:
--> 426 torch._C._cuda_emptyCache()
427
428

RuntimeError: CUDA error: device-side assert triggered