I opened the DQN tutorial in colab. And when I ran the following code in the “Input extraction” cell:
“”"
plt.figure()
plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, 0).numpy(),
       interpolation='none')

plt.title(‘Example extracted screen’)
“”".
I got this error. Anyone know how to fix it?