Remove unecessary code for synbn guard

pull/1317/head
Ross Wightman 2 years ago
parent 0da3c9ebbf
commit e27c16b8a0

@ -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

Loading…
Cancel
Save