README.md 252 Bytes
Newer Older
1
# Basic MNIST Example
2
3
4
5

```bash
python test_single_gpu_mnist.py
python test_single_gpu_mnist.py --use-te   # Linear layers from TransformerEngine
6
python test_single_gpu_mnist.py --use-te --use-fp8  # FP8 + TransformerEngine for Linear layers
7
```