1、DataLoader
data_loader = torch.utils.data.DataLoader(dataset, batch_size, num_workers, shuffle)
2、数据、模型需要放在cuda上
3、损失函数
torch.nn
4、优化器
torch.optim
pytorch易忘
原文:https://www.cnblogs.com/liujianing/p/12660564.html