Stack InvertedResidual blocks to build a layer for MobileNetV2. Args: out_channels (int): out_channels of block. num_

Pack all blocks in a stage into a ``ResLayer``.

make plugins for ResNet 'stage_idx'th stage . Currently we support to insert 'context_block', 'empirical_attention_block', 'nonlocal_

Converts mmsegmentation model (config + checkpoint) to TorchServe `.mar`. Args: config_file: In MMSegmentation config for

nn.Module: the normalization layer named "norm0"

nn.Module: the normalization layer named "norm1"

nn.Module: normalization layer after the first convolution layer

nn.Module: the normalization layer named "norm1"

nn.Module: normalization layer after the second convolution layer

nn.Module: normalization layer after the second convolution layer

Export Pytorch model to TorchScript model and verify the outputs are same between Pytorch and TorchScript. Args: model (nn.Module): P

Export Pytorch model to ONNX model and verify the outputs are same between Pytorch and ONNX. Args: model (nn.Module): Pytorch model w

This function disables printing when not in master process

Take integer y (tensor or variable) with n dims & convert it to 1-hot representation with n+1 dims.

Update center used for teacher output.

Update center used for teacher output.

Args: windows: (num_windows*B, window_size, window_size, C) H (int): Height of image W (int): Width of im