Class Config

model.py:62–71

@si_module
class TransformerVAE(nn.Module):
    class Config:
        io_config: Optional[GaussianMixtureIOLayer.Config] = None
        stack_config: Optional[Stack.Config] = None
        quantizer_config: Optional[LatentQuantizer.Config] = None

        plex_layer: int = None
        plex_roll: int = 1
        split: bool = True

        from_pretrained: Optional[Tuple[str, str]] = None

    def __init__(self, c: Config):
        super().__init__()