src/transformers/models/cvt/modeling_cvt.py/CvtEncoder/__init__
class CvtEncoder:    def __init__(self, config):
        super().__init__()
        self.config = config
        self.stages = nn.ModuleList([])
        for stage_idx in range(len(config.depth)):
            self.stages.append(CvtStage(config, stage_idx))