How to train models?

Hi,

I'm using Google Colaboratory for running the scripts, but when training models with the default settings, I get the error (along with output) attached below:
[https://pastebin.com/63vg3zKS](https://pastebin.com/63vg3zKS) 

Additionally, here is my notebook code so far
[notebook.zip](https://github.com/seaniezhao/torch_npss/files/7819415/notebook.zip)
:

Any help is appreciated!