Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- use_gpu = torch.cuda.is_available()
- inputs_try , labels_try = var_cgpu(inputs_try,use_gpu),var_cgpu(labels_try,use_gpu)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement