search

dobkeratops / convnet

2 stars 1 forks source link

An assignment is needed for the transfers to the device when CUDA is … #3

Open Twenkid opened 2 years ago

Twenkid commented 2 years ago

…used

data = data.to(device) target = target.to(device)

https://github.com/dobkeratops/convnet/issues/2#issuecomment-1264547041