Skip to content
#

onnx

Star

Here are 459 public repositories matching this topic...

Language: All
Filter by language
All 459 Python 217 Jupyter Notebook 80 C++ 49 C# 28 JavaScript 16 C 6 Go 5 Rust 5 Dockerfile 4 Java 4
Sort: Best match
Sort options
Best match Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated

onnx / onnx

Star 11.9k
Open

need tf.random.shuffle supported.

3
chan4cc
chan4cc commented Apr 26, 2021

New Operator

Describe the operator

Why is this operator necessary? What does it accomplish?

This is a frequently used operator in tensorflow/keras

Can this operator be constructed using existing onnx operators?

If so, why not add it as a function?

I don't know.

Is this operator used by any model currently? Which one?

Are you willing to contribute it?

Read more
operator good first issue enhancement
SynapseML

microsoft / SynapseML

Star 3k
Open

[LightGBM] Regression: how to penalize negative predictions

18
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?

Read more
enhancement good first issue LightGBM
Find more good first issues

666DZY666 / micronet

Star 1.6k

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

pytorch pruning convolutional-networks quantization xnor-net tensorrt model-compression bnn neuromorphic-computing group-convolution onnx network-in-network tensorrt-int8-python dorefa twn network-slimming integer-arithmetic-only quantization-aware-training post-training-quantization batch-normalization-fuse
  • Updated Oct 6, 2021
  • Python

PINTO0309 / PINTO_model_zoo

Sponsor
Star 1.3k

A repository that shares tuning results of trained models generated by TensorFlow / Keras. Post-training quantization (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization), Quantization-aware training. TensorFlow Lite. OpenVINO. CoreML. TensorFlow.js. TF-TRT. MediaPipe. ONNX. [.tflite,.h5,.pb,saved_model,tfjs,tftrt,mlmodel,.xml/.bin, .onnx]

caffe computer-vision model-zoo tensorflow model models keras pytorch pretrained-models coreml onnx tensorflow-lite tflite tensorflowjs tfjs openvino edgetpu mediapipe tflite-models tf-trt
  • Updated Jan 3, 2022
  • Python
AdvBox

advboxes / AdvBox

Star 1.2k

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.

security machine-learning deep-learning paddlepaddle adversarial-example adversarial-examples onnx fgsm adversarial-attacks deepfool graphpipe
  • Updated Sep 8, 2021
  • Jupyter Notebook

onnx / onnx-tensorflow

Star 1k
Open

Tensors in list passed to 'values' of 'ConcatV2' Op have types [float64, int64, int64, int64, int64] that don't all match.

6
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

Read more
good first issue
sparseml

neuralmagic / sparseml

Star 784

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

nlp sparsity tensorflow keras pytorch deep-learning-algorithms image-classification deep-learning-library pruning object-detection automl computer-vision-algorithms onnx deep-learning-models sparsification pruning-algorithms smaller-models model-sparsification sparsification-recipes recipe-driven-approaches
  • Updated Jan 3, 2022
  • Python

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