Common libraries for converting machine learning models to TVM Relay IR
Project description
Neo Framework -> Relay IR converters
This is a set of utility classes for converting ML models to TVM Relay IR.
Installation
This package does not assume any framework dependencies.
Usage
import tarfile
from neo_loader import load_model
model_artifacts = []
with tarfile.open('/path/to/model.tar.gz', 'r:gz') as tf:
tf.extractall()
model_artifacts = tf.getnames()
relay_artifacts = load_model(
model_artifacts=model_artifacts,
input_shape={'data': [1, 3, 224, 224]},
framework='tensorflow'
)
def relay_func(relay_artifacts: Dict) -> object:
return relay_artifacts['model_objects'][0]
def relay_params(relay_artifacts: Dict) -> object:
return relay_artifacts['model_objects'][1]
def relay_dtype(relay_artifacts: Dict) -> object:
if len(relay_artifacts['model_objects']) == 2:
return {}
else:
return relay_artifacts['model_objects'][2]
Security
See CONTRIBUTING for more information.
License
This project is licensed under the Apache-2.0 License.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for NeoCompilerModelLoaders-1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 768be4a78cd0f02f597a8e61d6d7c0887b072056f9f8d3836f33768ea526b236 |
|
MD5 | f638ad87fd22014e61934f4a8ec03353 |
|
BLAKE2b-256 | 60216c814af988b1b9ff832d7c70e455f8f01880fb5b36f2e39ad576ab6d4ddb |