Skip to content
#

onnx

Here are 597 public repositories matching this topic...

jcwchen
jcwchen commented Jun 1, 2022

Feature Request

System information

ONNX version (you are using): The latest main branch.

What is the problem that this feature solves?

To enhance robustness of node test data, ONNX CIs should have some ways to validate updated/uploaded input.pb/output.pb and ONNX models. Currently at least ONNX models have been covered by this PR: onnx/onnx#3855. However,

test good first issue enhancement build pipelines
SynapseML
brunocous
brunocous commented Sep 2, 2020

I have a simple regression task (using a LightGBMRegressor) where I want to penalize negative predictions more than positive ones. Is there a way to achieve this with the default regression LightGBM objectives (see https://lightgbm.readthedocs.io/en/latest/Parameters.html)? If not, is it somehow possible to define (many example for default LightGBM model) and pass a custom regression objective?

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

  • Updated Jun 3, 2022
  • Python

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape

  • Updated Oct 6, 2021
  • Python
AdvBox

Advbox is a toolbox to generate adversarial examples that fool neural networks in PaddlePaddle、PyTorch、Caffe2、MxNet、Keras、TensorFlow and Advbox can benchmark the robustness of machine learning models. Advbox give a command line tool to generate adversarial examples with Zero-Coding.

  • Updated May 26, 2022
  • Jupyter Notebook
morgoth95
morgoth95 commented Apr 24, 2022

The latest release of openvino has changed the inference engine API, providing a new API that takes advantage of the full potential of Intel's latest version of IR (IR 11). More information can be found at this link. We should adapt nebullvm's OpenVinoInferenceLearner to their latest API.

Usef

enhancement good first issue
ansh1204
ansh1204 commented Apr 27, 2020

I am trying to convert a custom pytorch model to tensorflow, I am abe to convert pytorch to onnx but converting onnx to tensorflow gives issue.

The code snippets are as follows-

pytorch to onnx

net = custom pytorch model
net.load_state_dict("pre-trained model")
dummyInput = np.random.uniform(0,1,(1,8,3,256,256))
dummyInput = Variable(torch.FloatTensor(dummyInput))
torch.onnx.export(ne

sparseml

Improve this page

Add a description, image, and links to the onnx topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the onnx topic, visit your repo's landing page and select "manage topics."

Learn more