技术心得(2.18-2.20)

1.pytorch模型提示超出内存cuda runtime error(2): out of memory

解决方案:https://ptorch.com/news/160.html

2.在数据读取过程中,如果出现数据维度前后不匹配的情况,该怎么办?

解决方案:(1)尝试将batch_size大小设为1,并且打印出每次所读取文件的路径。这种方案适用于DataLoader一次性读取全部文件的情况(2)将数据集转换为h5py文件,一次性读入。

发布了98 篇原创文章 · 获赞 5 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/chengsilin666/article/details/104369621
今日推荐