crack_segmentation icon indicating copy to clipboard operation
crack_segmentation copied to clipboard

Published 20 hours ago • verified publisher icon khanhha

in the utils.py

Open Chirag0096 opened this issue 1 year ago • 1 comments

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 avatar Sep 28 '23 05:09 Chirag0096