def __init__(self):
        self.required_inputs = []
        self.required_parameters = []

        self.inputs = {}
        self.parameters = {}

        settings = Settings()
        self.parameters = settings.get_network_parameters()
예제 #2
0
    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