mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
8 lines
210 B
8 lines
210 B
2 years ago
|
# Basic MNIST Example with optional FP8
|
||
|
|
|
||
|
|
```bash
|
||
|
|
python main.py
|
||
|
|
python main.py --use-te # Linear layers from TransformerEngine
|
||
|
|
python main.py --use-fp8 # FP8 + TransformerEngine for Linear layers
|
||
|
|
```
|