in the utils.py
Chirag0096 opened this issue · 1 comments
Chirag0096 commented
in line number 30
use this code to fix it
def cuda(x):
return x.cuda(non_blocking=True) if torch.cuda.is_available() else x
Chirag0096 commented
this will solve your issue which i was also facing and your work is appriciated man