From 45f17fc6ccb239b64c010831ddbcabe32984e4f3 Mon Sep 17 00:00:00 2001 From: duanjunwen <935724073@qq.com> Date: Thu, 26 Sep 2024 06:13:56 +0000 Subject: [PATCH] [fix] rm comments; --- tests/test_pipeline/test_schedule/test_zerobubble_pp.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/test_pipeline/test_schedule/test_zerobubble_pp.py b/tests/test_pipeline/test_schedule/test_zerobubble_pp.py index ccef295d4..46bd4a581 100644 --- a/tests/test_pipeline/test_schedule/test_zerobubble_pp.py +++ b/tests/test_pipeline/test_schedule/test_zerobubble_pp.py @@ -31,9 +31,6 @@ class MlpModel(nn.Module): ): super().__init__() self.layers = nn.Sequential(*[nn.Linear(in_dim, out_dim, bias=None) for _ in range(num_layers)]) - # self.layers = nn.ModuleList([nn.Linear(in_dim, out_dim, bias=None) for _ in range(num_layers)]) - # if stage_mgr: - # self.held_layers = self.layers[stage_index[0]:stage_index[1]] def forward( self,