parameters method

@override List<Tensor> parameters()

Implementation

@override
List<Tensor> parameters() {
  return [
    if (frameProjection != null) ...frameProjection!.parameters(),
    posEmbeddings, // Now this matches the field name above!
    ...transformerEncoder.parameters(),
    ...mlpHead.parameters(),
  ];
}