Build a Decoder

mmit.create_decoder(name, out_channels=None, out_reductions=None, **kwargs)

Build a decoder from a name and some keyword arguments.

Parameters:

• name (str) – The name of the decoder.

• out_channels (Optional[int]) – The number of channels of the input tensors of the forward pass.

• out_reductions (Optional[int]) – The reduction factor of the input tensors of the forward pass.

• kwargs – Keyword arguments for the decoder. Take a look at the specific decoder docs for more info!

Return type:

Module