But that would really imply that we have different archs altogether, right? And you are attempting to load the previous state_dict to this new arch by trimming the layer weights, let’s say?
What if your new arch has more neurons than the old one?