Solved by ptrblck in post #2 
              
              
                Do you get the same error, if you use strict=False in load_state_dict? 
This option should warn you, but ignore missing or unexpected keys.