Glad, I can help. Perhaps some intrinsic of pytorch deepcopy works differently in new version, the only reason I see.  Apparently, now all tensors “support the deepcopy protocol”, not only “graph leaves”