submodules method

@override List<Module> submodules()

Submodules owned by this module. Subclasses that compose other modules should override this so train() / eval() propagate. Default: empty.

Implementation

@override
List<Module> submodules() => [inputLn, postAttnLn, attn, ffn1, ffn2];