Module class abstract

Implementers

Constructors

Module()

Properties

hashCode int
The hash code for this object.
no setterinherited
runtimeType Type
A representation of the runtime type of the object.
no setterinherited
training bool
Whether this module is in training mode. Layers that behave differently between training and inference (e.g. Dropout) read this flag in their call method. Defaults to training mode.
getter/setter pair

Methods

eval() → void
Put this module (and any registered submodules) into evaluation mode.
noSuchMethod(Invocation invocation) → dynamic
Invoked when a nonexistent method or property is accessed.
inherited
parameters() List<Tensor>
Trainable tensors owned by this module (and its submodules).
submodules() List<Module>
Submodules owned by this module. Subclasses that compose other modules should override this so train() / eval() propagate. Default: empty.
toString() String
A string representation of this object.
inherited
train() → void
Put this module (and any registered submodules) into training mode.
zeroGrad() → void
Zero every parameter's gradient. Safe to call before each backward.

Operators

operator ==(Object other) bool
The equality operator.
inherited