I don’t know what the additional fp_ arguments mean, but would assume that the parameter loading should work as long as you don’t get any errors pointing to missing or unexpected keys in the state_dict.