parameters method

@override List<Value> parameters()

Implementation

@override
List<Value> parameters() {
  return [
    ...patchEmbedding.parameters(),
    ...stages.expand((s) => s.parameters()),
    ...finalNorm.parameters(),
    // if (classificationHead != null) ...classificationHead!.parameters(),
  ];
}