Skip to content

Instantly share code, notes, and snippets.

@vsay01

vsay01/create_model.py

Created January 22, 2020 02:00
Show Gist options
  • Save vsay01/114a0d427216dbaa041a898d81e09127 to your computer and use it in GitHub Desktop.
Save vsay01/114a0d427216dbaa041a898d81e09127 to your computer and use it in GitHub Desktop.
Download ZIP
Create instance model FashionClassifier and move to GPU if available
Raw
create_model.py
# Create the network, define the criterion and optimizer
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