AttributeError: 'list' object has no attribute 'item'

[eimalnuaimi]

Hey,
In #33, installing an older version of torchvision was the solution. Is there a work around to this issue that doesn't require installing older versions of torchvision? I use pytorch-nemo>=0.0.7, torch==1.9.0+cu102 and torchvision==0.10.0+cu102. Installing older versions of torchvision is off the table for me as I get persistent errors in essential segments of my code that I can't really play around with.