- pytorch-Dataloader多进程使用出错
此时在数据的调用之前加上if name == ‘main’:即可解决问题 - myset = TensorDataset(x_train,y_train) #这两个参数直接填进去,输入关键词会出错。
myloader = DataLoader(dataset=myset, batch_size=1, shuffle=False)
https://pytorch-cn.readthedocs.io/zh/latest/package_references/data/