2024年8月2日
torch.utils.data.Dataset 和 torch.utils.data.DataLoader
摘要： torch.utils.data是PyTorch中用于数据加载和预处理的模块。通常结合使用其中的Dataset和DataLoader两个类来加载和处理数据。 Dataset torch.utils.data.Dataset是一个抽象类，用于表示数据集。 需要用户自己实现两个方法：__len__和__ 阅读全文
posted @ 2024-08-02 21:23 华小电 阅读(130) 评论(0) 推荐(0) 编辑