How do I predict using using model trained on custom data using harvard pystruct?

Question

I am very new in pytorch. I was trying to train a model using custom data, using the following resources https://github.com/harvardnlp/pytorch-struct/blob/master/notebooks/BertD