Display pytorch matrices using matplotlib.

display_pytorch_matrics.py

import torch
import matplotlib.pyplot as plt
import numpy as np

# I learned this one in allennlp, hence the name.
p = "path/to/model/best.th"
w = torch.load(p)

for k in w.keys():
    print(k)
    t = w[k]
    try:    
        # Display matrix
        plt.matshow(t.cpu().numpy())
        plt.show()
    except Exception:
        # sometimes it fails.
        print("oops")