Struct NetworkHandle 

pub struct NetworkHandle {
    pub name: String,
    pub module: Option<PyObject>,
    pub optimizer: Option<PyObject>,
    pub scheduler: Option<PyObject>,
    pub batching: bool,
    pub k: Option<usize>,
    pub det: bool,
    pub train_mode: bool,
    pub cache_enabled: bool,
    pub cache_size: usize,
}

Handle to a registered neural network.

This struct holds the PyTorch module and associated training state. When the python feature is enabled, it can hold PyO3 PyObject references.

Fields

name: String

Unique name identifying this network

module: Option<PyObject>

The PyTorch nn.Module (set via Python API) Only available with the python feature

optimizer: Option<PyObject>

The optimizer for training (e.g., Adam, SGD) Only available with the python feature

scheduler: Option<PyObject>

Learning rate scheduler Only available with the python feature

batching: bool

Whether to batch inputs for efficient GPU processing

k: Option<usize>

Top-k sampling: if Some(k), only consider top k outputs

det: bool

Deterministic mode: use argmax instead of sampling

train_mode: bool

Whether the network is in training mode

cache_enabled: bool

Whether output caching is enabled

cache_size: usize

Maximum number of cached outputs

Implementations

impl NetworkHandle

pub fn new(name: String) -> Self

Create a new network handle with the given name and default settings.

pub fn from_config(config: &NetworkConfig) -> Self

Create a handle from a configuration.

pub fn has_module(&self) -> bool

Check if the PyTorch module has been set.

pub fn has_optimizer(&self) -> bool

Check if an optimizer has been configured.

pub fn has_scheduler(&self) -> bool

Check if a scheduler has been configured.

pub fn set_module(&mut self, module: PyObject)

Set the PyTorch module.

pub fn set_optimizer(&mut self, optimizer: PyObject)

Set the optimizer.

pub fn set_scheduler(&mut self, scheduler: PyObject)

Set the learning rate scheduler.

pub fn module(&self) -> Option<&PyObject>

Get a reference to the PyTorch module.

pub fn optimizer(&self) -> Option<&PyObject>

Get a reference to the optimizer.

pub fn scheduler(&self) -> Option<&PyObject>

Get a reference to the scheduler.

pub fn clear(&mut self)

Clear the module and training state.

Trait Implementations

impl Debug for NetworkHandle

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for NetworkHandle

fn default() -> Self

Returns the “default value” for a type.