pytorch easy to forget

1、DataLoader

data_loader = torch.utils.data.DataLoader(dataset, batch_size, num_workers, shuffle)

2, the data model needs to be placed on cuda

3, loss of function

torch.nn

4, the optimizer

torch.optim