org.pytorch

Class LiteModuleLoader

java.lang.Object

org.pytorch.LiteModuleLoader

public class LiteModuleLoader
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
LiteModuleLoader()

Method Summary

Modifier and Type	Method and Description
static Module	load(java.lang.String modelPath) Loads a serialized TorchScript module from the specified path on the disk to run on CPU.
static Module	load(java.lang.String modelPath, java.util.Map<java.lang.String,java.lang.String> extraFiles, Device device) Loads a serialized TorchScript module from the specified path on the disk to run on specified device.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LiteModuleLoader

public LiteModuleLoader()

Method Detail

load

public static Module load(java.lang.String modelPath,
                          java.util.Map<java.lang.String,java.lang.String> extraFiles,
                          Device device)

Loads a serialized TorchScript module from the specified path on the disk to run on specified device. The model should be generated from this api _save_for_lite_interpreter().

Parameters:

modelPath - path to file that contains the serialized TorchScript module.

extraFiles - map with extra files names as keys, content of them will be loaded to values.

device - Device to use for running specified module.

Returns:

new Module object which owns torch::jit::mobile::Module.

load

public static Module load(java.lang.String modelPath)

Loads a serialized TorchScript module from the specified path on the disk to run on CPU. The model should be generated from this api _save_for_lite_interpreter().

Parameters:

modelPath - path to file that contains the serialized TorchScript module.

Returns:

new Module object which owns torch::jit::mobile::Module.