添加num_workers=num,需要多设置一些,就可以使用GPU加速
train_loader = DataLoader(dataset=train_data, batch_size=train_bs, shuffle=True, num_workers=8)
添加num_workers=num,需要多设置一些,就可以使用GPU加速
train_loader = DataLoader(dataset=train_data, batch_size=train_bs, shuffle=True, num_workers=8)