Skip to content

Instantly share code, notes, and snippets.

@vsay01

vsay01/load_model_instance.py

Created January 22, 2020 12:59
Show Gist options
  • Save vsay01/16e62931ec8a62c2d46996703b867ebc to your computer and use it in GitHub Desktop.
Save vsay01/16e62931ec8a62c2d46996703b867ebc to your computer and use it in GitHub Desktop.
Download ZIP
Create load model instance
Raw
load_model_instance.py
model = FashionClassifier()
# move model to GPU if CUDA is available
if use_cuda:
model = model.cuda()
print(model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment