Get value by key at specified index (indices) if idx == None, returns value for key at each output index if idx is an integer, return

r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi

Build model with specified default_cfg and optional model_cfg This helper fn aids in the construction of a model including: * handling def

Creates an EfficientNet model. Ref impl: https://github.com/tensorflow/tpu/blob/master/models/official/efficientnet/efficientnet_model.py Pap

Select a 2d convolution implementation based on arguments Creates and returns one of torch.nn.Conv2d, Conv2dSame, MixedConv2d, or CondConv2d.

interleave 2 block types in stack

Feature map down-sampling.

Feature map interpolation.

Activation Layer Factory Fetching activation layers by name with this function allows export or torch script friendly functions to be returne

Scale from [0, 1] to [-1, +1]

Possibly convert HWIO to OIHW.

ResNet-V2 backbone helper

Creates a MobileNet-V3 model. Ref impl: ? Paper: https://arxiv.org/abs/1905.02244 Args: channel_multiplier: multiplier to number o

Creates an EfficientNet-Lite model. Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet/lite Paper: https://

Convert metric depth in [0, `max_depth`] to normalized depth in [0, 1].

Scale from [-1, +1] to [0, 1]