__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 |