当前标签: dimension

pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 a

<>1.pytorch报错 Namespace(batchSize=16, gpu='0,1', lr=0.001, nepoch=210, num_workers=8) Epoch: 0 Epoch:0 [0|173089] loss:3.747539 Epoch:0 [1|173089] loss:3.629...

2018-11-28 16:39 阅读(1954)

解决：Expected dimension in the range [-1, 1), but got 1

最近在学习TensorFlow的时候遇到了许多问题，现一一记录，帮助自己记忆的同时也希望能帮到大家。正文------------------------------------------------- 做分类任务的时候，经常会使用交叉熵函数作为loss来计算：形式如下 def evaluation(logit...

2018-09-28 15:24 阅读(1635)

1