Thank you so much @ptrblck! Well, I believe @Michela will know  Regarding what you mention, I will try then using state_dict. But it was already surprising to me that the size was changing that much, as on top of the weight_mask, I cannot think about any other overhead created on the pruned model.