From 5d37c1d053aff60e27856e1bba11555a5a2c4867 Mon Sep 17 00:00:00 2001
From: Jianghai <72591262+CjhHa1@users.noreply.github.com>
Date: Wed, 6 Sep 2023 12:02:39 +0800
Subject: [PATCH] Update sharder.py

---
 colossalai/shardformer/shard/sharder.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/colossalai/shardformer/shard/sharder.py b/colossalai/shardformer/shard/sharder.py
index 39704ae5e3ec..94ea8aa44f3b 100644
--- a/colossalai/shardformer/shard/sharder.py
+++ b/colossalai/shardformer/shard/sharder.py
@@ -28,7 +28,6 @@ class ModelSharder(object):
     def __init__(self, model: nn.Module, policy: Policy, shard_config: ShardConfig = None) -> None:
         self.model = model
         self.policy = get_autopolicy(self.model, shard_config.inference_only) if policy is None else policy
-        print(self.policy)
         self.shard_config = shard_config
 
     def shard(self) -> List[Dict[int, Tensor]]: