class_docstr: This model is to test with layers which go through forward pass multiple times.
  In this model, the fc1 and fc2 call forward twice
class_docstr_tok: ["This","model","is","to","test","with","layers","which","go","through","forward","pass","multiple","times",".","In","this","model",",","the","fc1","and","fc2","call","forward","twice"]
code: def __init__(self, checkpoint=False) -> None:
      super().__init__(checkpoint=checkpoint)
      self.fc1 = nn.Linear(5, 5)
      self.fc2 = nn.Linear(5, 5)
      self.fc3 = nn.Linear(5, 2)
      self.layers = [self.fc1, self.fc2, self.fc1, self.fc2, self.fc3]
  
file_path: ColossalAI/tests/kit/model_zoo/custom/repeated_computed_layers.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from ..registry import model_zoo
  from .base import CheckpointModule
label_desc: Use image node_id 1 to create a new NetWithRepeatedlyComputedLayers object from inherited base classes: CheckpointModule with example: obj = NetWithRepeatedlyComputedLayers(checkpoint)