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 物件。