Not seeing any issues. However, it’s a whole Module (well, a sub-module, whose parameters are added to the super-module’s registered parameter’s and the optimizer’s param_group, but is then later deleted), not a tensor.