I have done all the steps you’ve outlined, except possibly for the masking. I think I have tried masking the output, but perhaps it wasn’t very good. I’ll try again. Thank you for a really well written tutorial! This ought to be baked into the standard functions of pyTorch!