NVlabs / STEP

STEP: Spatio-Temporal Progressive Learning for Video Action Detection. CVPR'19 (Oral)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA error: an illegal memory access was encountered

guancheng817 opened this issue · comments

hi, thanks for you great works. When I add --fp16 flag, I meet a CUDA error: an illegal memory access was encountered error. However, when I remove the --fp16 flag, it works well. I have two gpus, one is GTX 1070 and another one is 980Ti.
pytorch version 1.1.0
cuda 9.0
thanks again..