I am struggling to work out why I’m getting this error when trying to apply the lottery ticket hypothesis to my model. Clearly this is happening during the pruning callback, and it seems like it’s trying to save the previous weights as a parameter, but the value of a FloatTensor. I am quite at a loss of what to do.