I also dont see it. I am using pytorch lightning so there are a lot of boilerplate code. This is the line it refers to. @soulitzer If you have any idea how to get that kind of error message let me know.