首页 推荐 热点 专题 PHPJavaPythonAndriodIOSC#前端数据库人工智能程序开发架构安全运维资讯微课资源
0
点赞
收藏
分享

微信扫一扫

深度学习:invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to

爪哇驿站 2022-04-05 阅读 78
迁移学习人工智能机器学习python图像处理

版本问题

源代码:

test_loss += F.nll_loss(out_tgt.log(), target_label, size_average=False).data[0] # sum up batch loss

修改后代码:

test_loss += F.nll_loss(out_tgt.log(), target_label, size_average=False).item()  # sum up batch loss
举报

相关推荐

TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to

desktoppython错误提示系统/运维

每天五分钟深度学习pytorch:tensor与python、numpy之间的转换

人工智能机器学习深度学习AI误判监控AI算法团队开发
0 条评论
爪哇驿站
关注
  • 04 搭建本地yum源
  • Pytest fixture测试用例前置、后置
  • 技术人必藏!解锁海外流量的隐形钥匙?实测90%跨境难题竟藏在这个底层细节
  • Harmonyos5应用开发实战——图片拼图功能实现(part2)
  • MOS管防反接:是用Nmos还是Pmos?
  • DeepSeek R1-32B医疗大模型的完整微调实战分析(全码版)
  • axios调用axis2错误
  • sql server删除注释
  • 重新面试之JVM
  • 为什么类 UNIX 操作系统通常内置编译器?为什么 Windows 更倾向于直接使用二进制文件?