U ¯Ê¹jVã@s4dZddlmZddlmZddlmZdd„ZdS)zQ Create Conv2d Factory Method Hacked together by / Copyright 2020 Ross Wightman é)Ú MixedConv2d)Ú CondConv2d)Úcreate_conv2d_padcKsÂt|tƒrXd|kst‚d|krF| d¡}||kr:d|d<n |dksFt‚t|||f|Ž}nf| dd¡}|rl|n | dd¡}d|kr¦|ddkr¦t|||fd|i|—Ž}nt|||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. Z num_expertsÚgroupsTÚ depthwiserFé)Ú isinstanceÚlistÚAssertionErrorÚpoprrr)Z in_channelsZ out_channelsZ kernel_sizeÚkwargsrÚmr©rúX/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/layers/create_conv2d.pyÚ create_conv2d s      rN)Ú__doc__Z mixed_conv2drZ cond_conv2drZ conv2d_samerrrrrrÚs