parameters method

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

Implementation

@override
List<Tensor> parameters() => [
  ...selfAttention.parameters(),
  ...norm1.parameters(),
  ...crossAttention.parameters(),
  ...norm2.parameters(),
  ...ff1.parameters(),
  ...ff2.parameters(),
  ...norm3.parameters(),
];