Unexpected key in state_dict: "bn1.num_batches_tracked"

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