Philipp

Windows 10 (x64) with Notepad++ v7.8.4 (64-bit)

As a workaround, simply remove the NppMarkdownPanel directory from your NPP plugins dir.

The problem seems to be the matplotlib, since cm.jet(...) is returning 4 jet images. So I just overlayed each individually. The 4th returning image seems to be the alpha channel...

@mimoralea OK, the closest I can get to the ```visualize_attention.ipynb``` narrative is to just select the first from the cm.jet(grads).  ``` # Lets overlay the heatmap onto original image....

Alright, thanks for the confirmation. ^^ I'll give a try and make sure that these tests run through https://github.com/Stable-Baselines-Team/stable-baselines3-contrib/blob/master/tests/test_lstm.py without any errors (and maybe even a bit faster).