ViTFaceEmbeddingGPU constructor

ViTFaceEmbeddingGPU({
  1. required int imageSize,
  2. required int patchSize,
  3. required int embedSize,
  4. int outputDim = 512,
  5. int numLayers = 4,
})

Implementation

ViTFaceEmbeddingGPU({
  required int imageSize,
  required int patchSize,
  required int embedSize,
  int outputDim = 512,
  int numLayers = 4,
}) : backbone = ViTBackbone(
       imageSize: imageSize,
       patchSize: patchSize,
       embedSize: embedSize,

       numLayers: numLayers,
     ) {
  if (embedSize != outputDim) {
    finalProjection = Layer(embedSize, outputDim, useGelu: false);
  } else {
    finalProjection = null;
  }
}