PyTorch 数据处理

PyTorch-相关笔记 2021-05-21 11:41:42 2021-05-21 11:41:42 590 次浏览

概述

Dataset

dataset = torch.utils.data.TensorDataset(features, labels)

Data Loader

data_iter = torch.utils.data.DataLoader(dataset, batch_size, shuffle=True)

注意,可以 num_workers 参数来使用多进行进行加载数据,可以提升训练速度。 但是对于 Windows,由于其不支持 fork 系统调用,因此会报错,解决方法:

if sys.platform.startswith('win'):
    num_workers = 0
else:
    num_workers = 4

参考

https://tangshusen.me/Dive-into-DL-PyTorch/#/chapter03_DL-basics/3.5_fashion-mnist

未经允许,禁止转载,本文源站链接:https://iamazing.cn/