1, DataLoader
data_loader = torch.utils.data.DataLoader (conjunto de datos, batch_size, num_workers, shuffle)
2, el modelo de datos tiene que ser colocado en cuda
3, pérdida de la función
torch.nn
4, el optimizador
torch.optim
pytorch fácil olvidar
Supongo que te gusta
Origin www.cnblogs.com/liujianing/p/12660564.html
Recomendado
Clasificación