@apaszke thanks for the hint, I tried that but it didn’t help. I’ll start fresh with a new model and see whether it goes wrong again. If it does, I’ll be closer to having a minimal test case to share.