o ÓÙ¾iVã@s4dZddlmZddlmZddlmZdd„ZdS)zQ Create Conv2d Factory Method Hacked together by / Copyright 2020 Ross Wightman é)Ú MixedConv2d)Ú CondConv2d)Úcreate_conv2d_padcKsÊt|tƒr/d|vs J‚d|vr#| d¡}||krd|d<n|dks#J‚t|||fi|¤Ž}|S| dd¡}|r9|n| dd¡}d|vrW|ddkrWt|||fd|i|¤Ž}|St|||fd|i|¤Ž}|S)zà Select a 2d convolution implementation based on arguments Creates and returns one of torch.nn.Conv2d, Conv2dSame, MixedConv2d, or CondConv2d. Used extensively by EfficientNet, MobileNetv3 and related networks. Ú num_expertsÚgroupsTÚ depthwiserFé)Ú isinstanceÚlistÚpoprrr)Ú in_channelsÚ out_channelsÚ kernel_sizeÚkwargsrÚmr©rúM/home/ubuntu/.local/lib/python3.10/site-packages/timm/layers/create_conv2d.pyÚ create_conv2d s      ùÿrN)Ú__doc__Ú mixed_conv2drÚ cond_conv2drÚ conv2d_samerrrrrrÚs