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() => [
  tokenEmb,
  posEmb,
  embedDrop,
  encoder,
  if (untiedHead != null) untiedHead!,
];