- 博客(3)
- 收藏
- 关注
RSS订阅原创 ValueError: Expected more than 1 value per channel when training, got input size torch.Size问题
修改数据集数量，让数据集数量除以batchsize余数为0或者大于1。后来问同门，在dataloder中可以设置如果除不尽直接舍弃余数。在跑模型的时候突然出现这个问题，我有很多数据集，模型到其中一个数据集的时候都没问题，就跑这个会有问题，以前没遇到过，今天尝试解决了。
2024-12-16 16:07:55
266
原创 Pycharm进行分布式训练
如果没有yml文件，就在程序中找Parameters，一般都写在option.py中。建错了就去文件目录下运行下面的代码，可以删除链接。在分布式中运行和调试PromptIR代码。
2023-12-06 11:20:55
379
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人