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.
evaluate_detection_average_precision
Calculate mAP for object detection results.
Available plugins for train task.
- log_summary
- log_tensorboard
- progressbar
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.2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4e96b0f2a481d26d0b5f22f95dbaa7f1829e5f1d27b729519dcc6f2da31d089a |
|
MD5 | fe25d09944f474a596961e43e9c85d10 |
|
BLAKE2b-256 | a4602910693d1696c3a06e7c38c10dfce8947bd7e6c01722de9269726fe2082f |
Close
Hashes for irisml_tasks_training-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c75cc184cedb9ef1eb07f9411d7e12e00b491559665881d9f411f15b3202c987 |
|
MD5 | 56c56921c71bf32db157065ec3bc79aa |
|
BLAKE2b-256 | d36d6255c3270bc55fc8cbec983f8e0e365b02692f65dfd5fd394823ec6dce36 |