Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/nlg_instantiate_model.py

Last active July 28, 2020 07:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prateekjoshi565/96d93426f74d43359a33fc6e35fe70c1 to your computer and use it in GitHub Desktop.
Save prateekjoshi565/96d93426f74d43359a33fc6e35fe70c1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nlg_instantiate_model.py
# instantiate the model
net = WordLSTM()
# push the model to GPU (avoid it if you are not using the GPU)
net.cuda()
print(net)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment