The strict argument should be passed to load_state_dict, not torch.load.
strict
load_state_dict
torch.load