From 06a3a100b330d10b28615af285642c6667ba8c23 Mon Sep 17 00:00:00 2001 From: genghaozhe <939857490@qq.com> Date: Fri, 17 May 2024 10:57:49 +0000 Subject: [PATCH] remove unrelated code --- colossalai/zero/gemini/placement_policy.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/colossalai/zero/gemini/placement_policy.py b/colossalai/zero/gemini/placement_policy.py index 4c3d8dbe2..c0d03ba3b 100644 --- a/colossalai/zero/gemini/placement_policy.py +++ b/colossalai/zero/gemini/placement_policy.py @@ -45,11 +45,6 @@ class PlacementPolicy(ABC): raise NotImplementedError -# import torch.distributed as dist -# # rank = int(os.environ["RANK"]) -# rank = dist.get_rank() - - class StaticPlacementPolicy(PlacementPolicy): def __init__( self,