Modules

Summary

module

truncated documentation

__init__

module onnxcustom Experimentation with ONNX, examples. source on GitHub

__init__

module onnxcustom.cli Shortcuts to cli. source on GitHub

__init__

module onnxcustom.plotting Shortcuts to plotting. source on GitHub

__init__

module onnxcustom.training Shortcuts to training. source on GitHub

__init__

module onnxcustom.utils Shortcuts to utils. source on GitHub

__main__

module onnxcustom.__main__ Implements command line python -m onnxcustom <command> <args>. source on GitHub

_base

module onnxcustom.training._base Base class for BaseEstimator and BaseOnnxFunction. source on GitHub

_base_estimator

module onnxcustom.training._base_estimator Optimizer with onnxruntime-training. source on GitHub

_base_onnx_function

module onnxcustom.training._base_onnx_function Helper for onnxruntime-training. source on GitHub

benchmark

module onnxcustom.utils.benchmark Tools to help benchmarking. source on GitHub

data_loader

module onnxcustom.training.data_loader Manipulate data for training. source on GitHub

doc_helper

module onnxcustom.utils.doc_helper Helpers to improve documentation rendering. source on GitHub

excs

module onnxcustom.training.excs Exceptions. source on GitHub

grad_helper

module onnxcustom.training.grad_helper ONNX and gradient. source on GitHub

imagenet_classes

module onnxcustom.utils.imagenet_classes Informations related to the ImageNet competition. source on GitHub

nvprof2json

module onnxcustom.utils.nvprof2json Converts traces from nvprof. The source comes from nvprof2json. source on GitHub

onnx_function

module onnxcustom.utils.onnx_function Onnx helper. source on GitHub

onnx_helper

module onnxcustom.utils.onnx_helper Onnx implementation of common functions used to train a model. source on GitHub

onnx_rewriter

module onnxcustom.utils.onnx_rewriter Rewrites operator in ONNX graph. source on GitHub

onnxruntime_helper

module onnxcustom.utils.onnxruntime_helper Onnxruntime helper. source on GitHub

optimizers

module onnxcustom.training.optimizers Optimizer with onnxruntime-training. source on GitHub

optimizers_partial

module onnxcustom.training.optimizers_partial Optimizer with onnxruntime-training forward backward training. source on GitHub

ortgradient

module onnxcustom.training.ortgradient Gradient with onnxruntime-training forward backward. source on GitHub

orttraining_helper

module onnxcustom.utils.orttraining_helper ONNX manipulations to help build ONNX gradient graphs. source on GitHub

plotting_onnx

module onnxcustom.plotting.plotting_onnx Shortcuts to plotting. source on GitHub

print_helper

module onnxcustom.utils.print_helper Helpers to display internal structures. source on GitHub

profiling

module onnxcustom.cli.profiling Command lines for profiling. source on GitHub

sgd_learning_loss

module onnxcustom.training.sgd_learning_loss Helper for onnxruntime-training. source on GitHub

sgd_learning_penalty

module onnxcustom.training.sgd_learning_penalty Helper for onnxruntime-training. source on GitHub

sgd_learning_rate

module onnxcustom.training.sgd_learning_rate Helper for onnxruntime-training. source on GitHub