Skip to content

Instantly share code, notes, and snippets.

@erykml

erykml/model_train.py

Created January 25, 2019 21:57
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save erykml/f2d368ff6232ab47e4db4f82acf3f76d to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save erykml/f2d368ff6232ab47e4db4f82acf3f76d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_train.py
model = train_cnn(model, train_loader, valid_loader,
criterion, optimizer, n_epochs = 30, train_on_gpu = True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment