I don’t understand the question.
Ofc your need layer changes state_dic, it has to appear there.
If the problem is you are having troubles loading the pretrained model try to use the argument strict=false  when loading the pretrained model. This allows you to load weights without having the exact order. Just use a singular name  for your new layers not get error while loading.