Unexpected key in state_dict: "bn1.num_batches_tracked"

Thank you this solved my problem