TorchLean API

Docs HomeGuideExamplesGraphs

NN.Examples.Models.Generative.VqVae

VQ-VAE-Style CIFAR Example #

Trains a small vector reconstruction model with a narrow tanh bottleneck, paired with the VQ-VAE spec/theory modules. The theorem-facing codebook objective lives in NN.Spec.Models.VqVae; this runtime example is the executable reconstruction path.

source
def NN.Examples.Models.Generative.VqVae.exeName :
String
Instances For
    source
    def NN.Examples.Models.Generative.VqVae.defaultLogJson :
    System.FilePath
    Instances For
      source
      def NN.Examples.Models.Generative.VqVae.cfg :
      API.nn.models.VectorGenerativeConfig
      Instances For
        source
        @[reducible, inline]
        abbrev NN.Examples.Models.Generative.VqVae.σ :
        Shape
        Instances For
          source
          @[reducible, inline]
          abbrev NN.Examples.Models.Generative.VqVae.τ :
          Shape
          Instances For
            source
            def NN.Examples.Models.Generative.VqVae.mkModel :
            API.nn.M (API.nn.Sequential σ τ)
            Instances For
              source
              def NN.Examples.Models.Generative.VqVae.main (args : List String) :
              IO UInt32
              Instances For