RuntimeError: Current CUDA Device does not support bfloat16. Please switch dtype to float16.

#32
by taoyang2 - opened

v100 RuntimeError: Current CUDA Device does not support bfloat16. Please switch dtype to float16.我该怎么处理????

V100 does not support bf16

change in modeling_GOT.py at line 581 bfloat to float

Sign up or log in to comment