k1D'
out_block_type: str = None
block_out_channels: typing.Tuple[int] = (32, 32, 64)
act_fn: str = None
norm_num_groups: int = 8
layers_per_block: int = 1
downsample_each_block: bool = False

)


Parameters 

sample_size (int, optional) — Default length of sample. Should be adaptable at runtime.


in_channels (int, optional, defaults to 2) — Number of channels in the input sample.


out_channels (int, optional, defaults to 2) — Number of channe