Spaces:
Running
on
T4
Running
on
T4
Help! What am I doing wrong?
#71
by
Stablovski
- opened
Please insert the following line before line 23 to ensure proper device selection:
device = "cuda" if torch.cuda.is_available() else "cpu"