Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- fn = FirstNetwork_v2()
- loss_fn = F.cross_entropy
- opt = optim.SGD(fn.parameters(), lr=1)
- fit_v2(X_train, Y_train, fn, opt, loss_fn)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
