def __init__(self): self.required_inputs = [] self.required_parameters = [] self.inputs = {} self.parameters = {} settings = Settings() self.parameters = settings.get_network_parameters()
def __init__(self, model, log_path): super().__init__() settings = Settings() self.training_parameters = settings.get_training_parameters() self.network_parameters = settings.get_network_parameters() self.log_path = log_path self.filename = '{}/{}'.format(log_path, 'config.txt') model.summary(print_fn=self._handle_summary_print) self.model = model