IrisML tasks for pytorch training
Project description
irisml-tasks-training
This is a package for IrisML training-related tasks.
Tasks
train
Train a pytorch model. A model object must have "criterion" and "predictor" property. See the documents for the detail. Returns a trained model.
predict
Run inference with a given pytorch model. Returns prediction results.
export_onnx
Trace a pytorch model and export it as ONNX using torch.onnx.export(). Throws an exception if it couldn't export. Returns an exported onnx model.
evaluate_accuracy
Calculate top1 accuracy for given prediction results. It supports only image classification results.
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
Built Distribution
Close
Hashes for irisml-tasks-training-0.0.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | e7bb64858f44cb7c4b7381123eacfd2afd6dbf8a35e13f79775bce11e8fb1730 |
|
MD5 | 5a1adeae8b6537cc7486ca4b7769d2c3 |
|
BLAKE2b-256 | 6849047bba8bb73801347dd22e2d6c45f743121f08047d0f2e12cff416161457 |
Close
Hashes for irisml_tasks_training-0.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d52947e8fd08ddedf410d448b0e80421b8f1c70ce32a489032e9a0510f12784a |
|
MD5 | 2d8962c76b9da953cbffb8e4cbc0213b |
|
BLAKE2b-256 | 1265aa16f48a9049c33e2729d443a4cb0686d95a84b4e3257604bf595c08f0bc |