-
**Environment:**
1. Framework: PyTorch 1.8.0
3. Horovod version: 0.21.3
4. MPI version:
5. CUDA version:
6. NCCL version:
7. Python version: 3.9
10. OS and version: Linux 5.11.10-arch1-1 #1 SMP…
-
when i ran this command
```
import onnx
from onnx_tf.backend import prepare
# Load the ONNX file
model = onnx.load('output/mnist.onnx')
# Import the ONNX model to Tensorflow
tf_rep = prepa…
-
We are working with FINN-ONNX, and we want the pretrained models from Brevitas that classify the MNIST images to output the index (class) instead of a probabilities tensor of dim 1x10.To our knowledge…
-
Hi. I'm trying using torch2trt_dynamic to convert models from torch to trt. I got an unexpected issue.
To reproduce it I use lenet5 like neural network.
```
from torch2trt_dynamic import torch2trt_…
-
Hi, I really like this project, and wonder if you can give me any advice on how to make a smaller architecture to train on MNIST?
In your paper on the last page, it's written that the `B-A-C-P` is…
-
Hi ZHAO Lanhuang, great work and very useful instruction for beginner.
I have question about the version of gcc-arm-none-eabi you use for this project.
In my computer, it is 9.2.1 running on the ubu…
-
Hi.
could you please provide test shell scripts for lenet, allconvnet probout.
Trying on my own I am unable to get it to work successfully.
Also Training with train_cifar_allconvnet_probout.sh, i…
-
```python
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
import tensorflow as tf
# model
x = tf.placeholder(tf.fl…
-
## Issue description
All the other samplers for `torch.utils.data.DataLoader()` are designed to be used such that you can iterate an epoch (one epoch being the number of listed indices/data-points)…
-
Hi there,
I am currently trying to determine the robustness of sparse networks against adversarial examples. For this we are trying to reproduce the results from your paper; specifically the MNIST-…