[fix] rm comments;

2024-09-26 06:13:56 +00:00 · 2024-09-26 06:13:56 +00:00 · 45f17fc6cc
parent a92e16719b
commit 45f17fc6cc
1 changed files with 0 additions and 3 deletions
--- 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,