def load_state_dict(self, state_dict, **kwargs):
    state_dict.pop('fc.bias')
    state_dict.pop('fc.weight')
    super().load_state_dict(state_dict, **kwargs)
def load_state_dict(self, state_dict, **kwargs):
    state_dict.pop('fc.bias')
    state_dict.pop('fc.weight')
    super().load_state_dict(state_dict, **kwargs)