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
pythonfeature is enabled - Optional optimizer for training
- Optional learning rate scheduler
- Configuration flags for batching, caching, etc.
Structsยง
- Embedding
Handle - Handle to a registered embedding module.
- Network
Handle - Handle to a registered neural network.