The saved file only contains the state dict. So only the dictionary of parameters. You won’t find the layers parameters there, unless you save them explicitly on the side.
If you can load the layer informations before and create the PrunedNetwork with the right size, then it might be easier to load it afterwards.