2020年9月21日
pytorch深度学习：一般分类器
摘要： 使用的criterion不是MSE而是交叉熵。 numpy.shape，tensor.size()，正确遍历变量。 另外 CrossEntropyLoss的参数真是有够饶人的。 1 import torch 2 from torch import nn,optim 3 import matplotl 阅读全文
posted @ 2020-09-21 10:24 Lovaer 阅读(443) 评论(0) 推荐(0) 编辑