File size: 481 Bytes
3261444 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Muon Optimizer Test
This directory contains a test script for the **Muon optimizer**.
To execute the test, simply run:
```bash
# By default, the test will use 8 GPUs.
./run_test.sh
```
The number of GPUs can be controlled with the NGPU environment variable.
For example, to run with 4 GPUs:
```bash
NGPU=4 ./run_test.sh
```
## Limitations:
- Multi-node execution is not supported yet.
- Ensure that the specified number of GPUs is available on your machine before running.
|