改写torch.utils.data.Dataset的过程中,总结了几个个教训:
1.不要用Series数据创建csv文件,否则再度读取该文件会把首行数据作为columns标题,使得数据无法转为tensor格式;
2.改写Dataset时,一定不要把__getitem__()写成__getitems__()或其他内容,否则按照索引(例如Data[0])读取时会失败(但调用改写的函数读取数据__getitems__(0)不会出错)。
自定义数据集MyDataset过程经验记录
最新推荐文章于 2023-08-03 21:43:23 发布