Module io

Source
Expand description

Saving and loading of tensors from disk

Traits§

ModuleIO
This trait is implemented automatically for all modules that implement IntoIterator<Item = &mut Tensor>

Functions§

load
Load all parameters from file
save
Save all tensors into file. All parameters must be realized before calling this function, otherwise it will panic.