DonutConfig.small constructor

DonutConfig.small()

Configuration for a small model (for testing/development).

Implementation

factory DonutConfig.small() => const DonutConfig(
      inputSize: [640, 480],
      windowSize: 5,
      encoderLayer: [2, 2, 6, 2],
      decoderLayer: 2,
      maxLength: 256,
      encoderEmbedDim: 64,
      encoderNumHeads: [2, 4, 8, 16],
      decoderEmbedDim: 256,
      decoderFfnDim: 1024,
      decoderNumHeads: 8,
      vocabSize: 10000,
    );