将pytorch从0.3.0升级到1.0.0出现。问题同样适用于0.3.0-0.4.0升级后。
经过检查是版本兼容的问题,loss函数定义出错了。
之前是:
losses.append(float(loss.cpu().data.numpy()))[0]
修改为:
losses.append(float(loss.cpu().data.numpy()))
问题完美解决,记录一下。
将pytorch从0.3.0升级到1.0.0出现。问题同样适用于0.3.0-0.4.0升级后。
经过检查是版本兼容的问题,loss函数定义出错了。
之前是:
losses.append(float(loss.cpu().data.numpy()))[0]
修改为:
losses.append(float(loss.cpu().data.numpy()))
问题完美解决,记录一下。