From 5cc84d94dca1b3e3e4d162c1036b1fd496aa2a38 Mon Sep 17 00:00:00 2001
From: ver217 <lhx0217@gmail.com>
Date: Thu, 3 Mar 2022 15:55:27 +0800
Subject: [PATCH] rename shared adam to sharded optim v2

---
 .../zero/sharded_optim/{sharded_adam.py => sharded_optim_v2.py} | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
 rename colossalai/zero/sharded_optim/{sharded_adam.py => sharded_optim_v2.py} (99%)

diff --git a/colossalai/zero/sharded_optim/sharded_adam.py b/colossalai/zero/sharded_optim/sharded_optim_v2.py
similarity index 99%
rename from colossalai/zero/sharded_optim/sharded_adam.py
rename to colossalai/zero/sharded_optim/sharded_optim_v2.py
index 52b0bf5f7..dc8b48db8 100644
--- a/colossalai/zero/sharded_optim/sharded_adam.py
+++ b/colossalai/zero/sharded_optim/sharded_optim_v2.py
@@ -22,7 +22,7 @@ class OptimState(Enum):
     UNSCALED = 2
 
 
-class ShardedAdam(ColossalaiOptimizer):
+class ShardedOptimizerV2(ColossalaiOptimizer):
 
     def __init__(self,
                  adam_optim: Optimizer,