bitorch.models.common_layers.get_initial_layers

bitorch.models.common_layers.get_initial_layers(variant: Optional[Union[List[int], str]], input_channels: int, output_channels: int) → List[Module]

returns the initial layers for the given variant