@ptrblck Thanks for your comment, I was aware of it being Python3.10-related but I thought I should ask here in case there are any insights on how to solve this, or even whether there’s a “better” way to parallelize my model.
Indeed, with python 3.9 I had no problems (not tested with python 3.9 AND PyTorch 1.11 though).