def METHOD_NAME(self, state_dict, prefix, local_metadata, strict,
                          missing_keys, unexpected_keys, error_msgs):
    num_batches_tracked_key = prefix + 'num_batches_tracked'
    if num_batches_tracked_key in state_dict:
        del state_dict[num_batches_tracked_key]
    super(FrozenBatchNorm2d, self).METHOD_NAME(
        state_dict, prefix, local_metadata, strict,
        missing_keys, unexpected_keys, error_msgs)