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.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | dd94e0e01d9410156502770a02367e07b94bc9761d969b20d7ceb058b4ac1679 |
|
MD5 | 80120f9f8bfb5126390e733e8868e96f |
|
BLAKE2b-256 | e831129a084409f79a5e4f3f0fb356f15d57e85cb17fd89acb0e515d68dd64c9 |
Close
Hashes for irisml_tasks_training-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3835e59b80d646508e83228d655218eff854179302c21a94d685b19553a2440b |
|
MD5 | a832a80ead38a9336c690a7b6a7303b9 |
|
BLAKE2b-256 | 5d5cf64905e16e9961d1bf9e207180647d21d2bf641b7484bb4af96b72e737db |