需要把inputs和labels 放到cuda上
for batch, (inputs, labels) in enumerate(dataloader):
+ inputs = inputs.to(device)
+ labels = labels.to(device)
👍 说明你真跑了
Sign in to make a reply
mzzzZ09
需要把inputs和labels 放到cuda上
for batch, (inputs, labels) in enumerate(dataloader):
+ inputs = inputs.to(device)
+ labels = labels.to(device)