Translations:PyTorch/189/fr

From Alliance Doc
Jump to navigation Jump to search

criterion = nn.CrossEntropyLoss()

   optimizer = optim.SGD(net.parameters(), lr=args.lr)