parameters method

  1. @override
List<Tensor> parameters()
override

Implementation

@override
List<Tensor> parameters() => [
  ...featureProjection.parameters(),
  posEmbeddings,
  ...transformerEncoder.parameters(),
  ...classificationHead.parameters(),
];