Skip to main content

Module handle

Module handle 

Source
Expand description

Network handle for managing PyTorch modules.

Each registered neural network is represented by a NetworkHandle which holds:

  • The PyTorch module (nn.Module) - when python feature is enabled
  • Optional optimizer for training
  • Optional learning rate scheduler
  • Configuration flags for batching, caching, etc.

Structsยง

EmbeddingHandle
Handle to a registered embedding module.
NetworkHandle
Handle to a registered neural network.