parameters method

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

Implementation

@override
List<Value> parameters() {
  return [
    ...patchProjection.parameters(),
    ...clsToken.values, // CLS token itself is a parameter
    ...positionEmbeddings.expand((vec) => vec.values),
    ...transformerEncoder.parameters(),
  ];
}