折腾了一上午,看到别人写的博客没办法解决我的问题。忽然想到灵感。
我要做的是,测试一下我自己写的数据是否可以通过网络输出。但只有一个批量。
他会报如下错误

这里只需要把model状态改成eval()模式就可以了,因为在train()状态下是进行批量训练的。
改之后的代码为

折腾了一上午,看到别人写的博客没办法解决我的问题。忽然想到灵感。
我要做的是,测试一下我自己写的数据是否可以通过网络输出。但只有一个批量。
他会报如下错误

这里只需要把model状态改成eval()模式就可以了,因为在train()状态下是进行批量训练的。
改之后的代码为

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。