- 博客(2)
- 收藏
- 关注
原创 TypeError: forward() missing 2 required positional arguments: ‘input‘ and ‘label‘
【代码】TypeError: forward() missing 2 required positional arguments: ‘input‘ and ‘label‘
2023-06-07 20:15:24
730
原创 复现CDCGAN训练MNIST是遇到的问题
而对于tensor的计算操作,默认是要进行计算图的构建的,在这种情况下,可以使3.用 with torch.no_grad():,强制之后的内容不进行计算图构建。在torch.no_grad() 会影响pytorch的反向传播机制,在测试时因为确定不会使用到反向传播因此 这种模式可以帮助节省内存空间。解决:transform里面的归一化吧[(0.5),(0.5),(0.5)]都改成[(0.5)]就行了,因为MNIST数据集是黑白图像,这里用tansform进行了3个通道的归一化,所以错了。
2023-06-07 20:14:01
340
空空如也
YOLOv5锚框改进
2024-05-15
求分享绝缘子缺陷检测的数据集
2024-05-08
训练YOLO模型,怎么绘制与其他模型的对比图
2023-08-01
北京大学发布的PCB瑕疵数据集怎么使用
2023-07-06
在Jupyter lab里运行代码的时候怎么看代码是否运行完毕
2023-03-27
modbus通讯可以用什么开发板
2022-11-15
TA创建的收藏夹 TA关注的收藏夹
TA关注的人