PyTorch 2 · Training and saving
Choose a loss and optimizer
Set up cross-entropy loss and the Adam optimizer.
loss_fn = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=1e-3)
Official PyTorch 2 documentation
PyTorch 2 · Training and saving
Set up cross-entropy loss and the Adam optimizer.
loss_fn = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=1e-3)
Official PyTorch 2 documentation
Was this page helpful?