def METHOD_NAME(self, mode: bool = True):
    assert (
        not mode or self.build_residual_branches
    ), "Trying to train a model without residual branches, set arch_params.build_residual_branches to True and retrain the model"
    super(RepVGG, self).METHOD_NAME(mode=mode)