Adityanagraj/torch.py

Created July 2, 2020 08:57

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/Adityanagraj/544f786de1510f9811728e4d7d8a6591.js"></script>
Save Adityanagraj/544f786de1510f9811728e4d7d8a6591 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

	model = FishModel()
	epochs = 1000
	lr = 1e-2
	history1 = fit(epochs, lr, model, train_loader, val_loader)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment