改写torch.utils.data.Dataset的过程中,总结了几个个教训:
1.不要用Series数据创建csv文件,否则再度读取该文件会把首行数据作为columns标题,使得数据无法转为tensor格式;
2.改写Dataset时,一定不要把__getitem__()写成__getitems__()或其他内容,否则按照索引(例如Data[0])读取时会失败(但调用改写的函数读取数据__getitems__(0)不会出错)。
自定义数据集MyDataset过程经验记录
最新推荐文章于 2026-05-03 06:37:25 发布
本文分享了在改写PyTorch的torch.utils.data.Dataset过程中遇到的常见问题及解决方案,包括使用Series数据创建csv文件时的注意事项,以及正确实现__getitem__()方法的重要性。

1533

被折叠的 条评论
为什么被折叠?



