Ok,
In that case, I would use for mod_uniq_name, mod in model.named_modules() to find all the Linear layers and save their weights in some structure (like a dict) using mod_uniq_name as the key.
Then when you want to reload it, do the same iteration and if the mod_uniq_name is present in your saved structure, load what is there.