Ah okey. Thank you very much for the info and all your help in the forums!
What issue should I submit? Upload traced model or the model definition code?
I have an issue open for None objects dissapearing fron inside nn.ModuleList