torch (version 0.14.1)

jit_serialize: Serialize a Script Module

Description

Serializes a script module and returns it as a raw vector. You can read the object again using jit_unserialize.

Usage

jit_serialize(obj)

Value

raw()

Arguments

obj

(script_module)
Model to be serialized.

Examples

if (torch_is_installed()) {
model <- jit_trace(nn_linear(1, 1), torch_randn(1))
serialized <- jit_serialize(model)
}