2021年8月12日
torch中DataLoader加载字典
摘要： 参考资料： https://pytorch.org/docs/stable/data.html 我们都知道，Pytorch加载数据的常规流程是先定义一个Dataset，然后使用DataLoader来成Batch地加载数据。当我们在Dataset中定义的__getitem__返回的数据是array或者 阅读全文
posted @ 2021-08-12 11:44 思念殇千寻 阅读(967) 评论(0) 推荐(0)