The definition of the load_state_dict() in the source code is
def load_state_dict(self, state_dict, strict=True):
However, when I use it by “model.load_state_dict(pretrain_model, False)” , the error is “TypeError: load_state_dict() takes exactly 2 arguments (3 given)”
Can anyone tell me why it happen ?