Package org.pytorch

Class LitePyTorchAndroid

java.lang.Object

org.pytorch.LitePyTorchAndroid

public final class LitePyTorchAndroid
extends Object

Constructor Summary

Constructors

Constructor	Description
LitePyTorchAndroid()

Method Summary

Modifier and Type	Method	Description
static Module	loadModuleFromAsset(AssetManager assetManager, String assetName)
static Module	loadModuleFromAsset(AssetManager assetManager, String assetName, Device device)	Attention: This is not recommended way of loading production modules, as prepackaged assets increase apk size etc.
static void	setNumThreads(int numThreads)	Globally sets the number of threads used on native side.

Methods inherited from class java.lang.Object

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

Constructor Detail

LitePyTorchAndroid

public LitePyTorchAndroid()

Method Detail

loadModuleFromAsset

public static Module loadModuleFromAsset(AssetManager assetManager,
                                         String assetName,
                                         Device device)

Attention: This is not recommended way of loading production modules, as prepackaged assets increase apk size etc. For production usage consider using loading from file on the disk Module.load(String).

This method is meant to use in tests and demos.

loadModuleFromAsset

public static Module loadModuleFromAsset(AssetManager assetManager,
                                         String assetName)

setNumThreads

public static void setNumThreads(int numThreads)

Globally sets the number of threads used on native side. Attention: Has global effect, all modules use one thread pool with specified number of threads.

Parameters:

numThreads - number of threads, must be positive number.