Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
- loss_function = nn.CrossEntropyLoss()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement