Solved by rahulvigneswaran in post #2 
              
              
                Some aspects of matplotlib will try to convert the input you provide it into numpy. It can do it only if the provided input tensor is in cpu. 
So, to begin with I would suggest you to try and do .cpu() to the inputs that you provide, that is, check if every value in the dict history that you are usi…