def __init__(self, exp_config): model = create_base_model(exp_config) super().__init__(model, exp_config) num_parameters = np.sum( [p.numel() for p in self.main_model.parameters()]) logger.info(f"The model has {num_parameters} parameters.") raise Exception("Done")
def __init__(self, exp_config): model = create_base_model(exp_config) super().__init__(model, exp_config)
def __init__(self, exp_config): self._load_config(exp_config) model = create_base_model(exp_config) super().__init__(model, exp_config)