1.pytorch模型提示超出内存cuda runtime error(2): out of memory
解决方案:https://ptorch.com/news/160.html
2.在数据读取过程中,如果出现数据维度前后不匹配的情况,该怎么办?
解决方案:(1)尝试将batch_size大小设为1,并且打印出每次所读取文件的路径。这种方案适用于DataLoader一次性读取全部文件的情况(2)将数据集转换为h5py文件,一次性读入。
1.pytorch模型提示超出内存cuda runtime error(2): out of memory
解决方案:https://ptorch.com/news/160.html
2.在数据读取过程中,如果出现数据维度前后不匹配的情况,该怎么办?
解决方案:(1)尝试将batch_size大小设为1,并且打印出每次所读取文件的路径。这种方案适用于DataLoader一次性读取全部文件的情况(2)将数据集转换为h5py文件,一次性读入。