diff --git a/timm/models/layers/norm_act.py b/timm/models/layers/norm_act.py index 261bdb0a..ea5b7883 100644 --- a/timm/models/layers/norm_act.py +++ b/timm/models/layers/norm_act.py @@ -5,11 +5,6 @@ from typing import Union, List, Optional, Any import torch from torch import nn as nn from torch.nn import functional as F -try: - from torch.nn.modules._functions import SyncBatchNorm as sync_batch_norm - FULL_SYNC_BN = True -except ImportError: - FULL_SYNC_BN = False from .trace_utils import _assert from .create_act import get_act_layer