会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Halo辉Go
博客园
首页
新随笔
联系
订阅
管理
2021年10月12日
pytorch中遇到Expected more than 1 value per channel when training, got input size torch.Size([1, 128, 1, 1])解决方法。
摘要: 折腾了一上午,看到别人写的博客没办法解决我的问题。忽然想到灵感。 我要做的是,测试一下我自己写的数据是否可以通过网络输出。但只有一个批量。 他会报如下错误 这里只需要把model状态改成eval()模式就可以了,因为在train()状态下是进行批量训练的。 改之后的代码为
阅读全文
posted @ 2021-10-12 12:55 Halo辉Go
阅读(1720)
评论(0)
推荐(0)
编辑
公告