jit_serialize function

Serialize a Script Module

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

jit_serialize(obj)

Arguments

• obj: (script_module)

  Model to be serialized.

Returns

raw()

Examples

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