Struct tch::jit::TrainableCModule [−][src]
pub struct TrainableCModule { /* fields omitted */ }
Expand description
The trainable version of a jit PyTorch module.
These modules can be created via the TorchScript python api.
Implementations
Loads a PyTorch saved JIT module from a file.
This function also adds the tensors from the JIT module to the VarStore path passed as argument so that the module can be trained.
Loads a PyTorch saved JIT model from a read instance.
This function also adds the tensors from the JIT module to the VarStore path passed as argument so that the module can be trained.
Switches the module to training mode.
Switches the module to evaluation mode.
Performs the forward pass for a model on some specified tensor inputs.
Performs the forward pass for a model on some specified ivalue inputs.
Runs a specified entry point for a model on some given tensor inputs.