- 博客(2)
- 收藏
- 关注
RSS订阅原创 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
1046
原创 复现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
464
1
空空如也
在训练yolov5的时候，为什么明明显示成功调用gpu,但是实际GPU并没有在工作
2024-09-13
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关注的人