mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Remove unecessary code for synbn guard

fixes-syncbn_pretrain_cfg_resolve
Ross Wightman 2022-06-24 14:57:42 -07:00
parent 0da3c9ebbf
commit e27c16b8a0
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/layers/norm_act.py
View File

@ -5,11 +5,6 @@ from typing import Union, List, Optional, Any
import torch import torch
from torch import nn as nn from torch import nn as nn
from torch.nn import functional as F 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 .trace_utils import _assert
from .create_act import get_act_layer from .create_act import get_act_layer