PyTorch 数据处理

    
        标签：
        
            PyTorch 相关笔记
        
    发布于：2021-05-21 11:41:42
    编辑于：2021-05-21 11:41:42
    浏览量：1355

概述

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

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