From 45c32fffc4fdd04b0d35a14598e43deb1bf5d808 Mon Sep 17 00:00:00 2001 From: skydoorkai Date: Thu, 23 Jan 2025 11:55:29 +0800 Subject: [PATCH] fix link --- README.md | 3 ++- atorch/local_sgd/README.md | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 7f2a74c..3ef0a81 100644 --- a/README.md +++ b/README.md @@ -43,6 +43,7 @@ ATorch is an extension library of PyTorch developed by Ant Group's AI Infrastruc * Transformer ops * High-performance MOE * sub-graph compilation + * [Local SGD/EDiT](docs/README-EDiT.md) * Checkpointing * Mixed precision * Communication optimization @@ -109,7 +110,7 @@ python -m atorch.distributed.run --nproc_per_node 2 train.py --model_type llam [WSAM optimizer](docs/README-WSAM.md) - +[Local SGD/EDiT](docs/README-EDiT.md) ## Contributing diff --git a/atorch/local_sgd/README.md b/atorch/local_sgd/README.md index a58a16c..d373529 100644 --- a/atorch/local_sgd/README.md +++ b/atorch/local_sgd/README.md @@ -1,3 +1,3 @@ # Unified Local SGD (EDiT) Support -We atorch, we support local sgd (EDiT) integration with mainstream parallelism strategies. For detail, refer to [EDiT](../../docs/README-EDIT.md) \ No newline at end of file +We atorch, we support local sgd (EDiT) integration with mainstream parallelism strategies. For detail, refer to [EDiT](../../docs/README-EDiT.md)