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]]: