org.pytorch

类 LiteModuleLoader

java.lang.Object

org.pytorch.LiteModuleLoader

public class LiteModuleLoader
extends java.lang.Object

构造方法摘要

构造方法

构造方法及说明
LiteModuleLoader()

方法摘要

修饰符和类型	方法及说明
static Module	load(java.lang.String modelPath) 从磁盘上指定路径加载序列化 TorchScript 模块，以便在 CPU 上运行。
static Module	load(java.lang.String modelPath, java.util.Map<java.lang.String,java.lang.String> extraFiles, Device device) 从磁盘上指定路径加载序列化 TorchScript 模块，以便在指定设备上运行。

从类 java.lang.Object 继承的方法

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

构造方法详情

LiteModuleLoader

public LiteModuleLoader()

方法详情

load

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

从磁盘上指定路径加载序列化 TorchScript 模块，以便在指定设备上运行。该模型应通过此 API _save_for_lite_interpreter() 生成。

参数

modelPath - 包含序列化 TorchScript 模块的文件路径。

extraFiles - 具有额外文件名称作为键的映射，它们的内容将加载到值中。

device - 用于运行指定模块的 Device。

返回

拥有 torch::jit::mobile::Module 的新的 Module 对象。

load

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

从磁盘上指定路径加载序列化 TorchScript 模块，以便在 CPU 上运行。该模型应通过此 API _save_for_lite_interpreter() 生成。

参数

modelPath - 包含序列化 TorchScript 模块的文件路径。

返回

拥有 torch::jit::mobile::Module 的新的 Module 对象。