2020年5月4日
torch.utils.data.DataLoader
摘要： DataLoader 它是PyTorch中数据读取的一个重要接口，该接口定义在dataloader.py中，只要是用PyTorch来训练模型基本都会用到该接口（除非用户重写…），该接口的目的：将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小 阅读全文
posted @ 2020-05-04 22:23 OliYoung 阅读(404) 评论(0) 推荐(0)