adjust_conv_channels

adjust_conv_channels(old_conv: torch.nn.Conv2d, in_channels: int, pretrained: bool = True) → Union[torch.nn.Conv2d, torch.nn.Sequential]

Parameters

• old_conv (torch.nn.Conv2d) –

• in_channels (int) –

• pretrained (bool) –

Return type

Union[torch.nn.Conv2d, torch.nn.Sequential]