-
The setup:
Basic MNIST model as proof of concept using pytorch lightning with xla libraries on top of the Neuron DLAMI. Training on 2 Neuron cores (trn1.2xlarge) works well and successfully calls t…
-
Hi all,
I would like to know whether the nn_quantizer.py and code_gen.py works for all caffe models with specified restrictions.
As an example:
1. In NXP mnist example, they are mentioning lik…
-
Hi @Jerry-Ge ,
I have run the https://github.com/pytorch/executorch/blob/main/examples/arm/run.sh example done and success, now I am try to modify it to run a quantize int8 pytorch model which nee…
-
### Bug description
Running the [mnist-tutorial](https://github.com/Lightning-AI/tutorials/blob/publication/.notebooks/lightning_examples/mnist-tpu-training.ipynb) from Lightning-AI doesn't create …
-
### 1. System information
- Occurs in Google Colab w/ TF 2.14
- Have also verified w. TF 2.7 (Anaconda) on Windows 10
### 2. Code
[Colab to reproduce issue](https://colab.research.google.com…
-
In PyTorch, the underlying tensor operations that perform convolution and max pooling are provided by specific functions and modules. Let's look at each in detail:
### Convolution
The convolutio…
-
Looks like the GPU in colab is not being engaged. Tried using A100, V100, T4 GPU, and TPU hardware settings in colab. command:
`python train.py spacetimeformer mnist --embed_method spatio-temporal …
-
Hello! I found out that the MNIST dataset is loaded with one .NPZ file while training NN!
But output of this code is very strange! Why it 4 output file? How can i load my own dataset it to NN??
Wher…
-
## 🐛 Bug
When writing predictions with a `torch.save` together with a `BasePredictionWriter` (see [this example](https://pytorch-lightning.readthedocs.io/en/stable/extensions/generated/pytorch_ligh…
-
For part 4/4 of section 8, the tutorial says to use:
cross_entropy = tf.nn.softmax_cross_entropy_with_logits(Ylogits, Y_)
However, when you use this function you must use named arguments (or…