2024年3月1日
自定义 DataLoader 时应使用 Unix 系统
摘要： 自定义 Dataset 类 PyTorch 允许自定义 Dataset 类，并由此获得 DataLoader，能方便训练时获得 batch： from torch.utils.data import DataLoader, Dataset import h5py import os class Ra 阅读全文
posted @ 2024-03-01 21:22 倒地 阅读(25) 评论(0) 推荐(0)