Skip to content

Instantly share code, notes, and snippets.

@vaibhavkumar049

vaibhavkumar049/f22.py

Created June 8, 2019 20:05
Show Gist options
  • Save vaibhavkumar049/5c0e76724cb8062123da963420cf0ed9 to your computer and use it in GitHub Desktop.
Save vaibhavkumar049/5c0e76724cb8062123da963420cf0ed9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
f22.py
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)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment