search

ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。
http://tangshusen.me/Dive-into-DL-PyTorch
Apache License 2.0
18.25k stars 5.39k forks source link

5.5LeNet 非错误,初学者的一个小疑问 #145

Closed fjd330621 closed 4 years ago

fjd330621 commented 4 years ago

bug描述 train_ch5函数中 train_l_sum += l.cpu().item() train_acc_sum += (y_hat.argmax(dim=1) == y).sum().cpu().item()

为什么需要将误差建立在cpu上

版本信息 pytorch: torchvision: torchtext: ...