import tensorflow as tf

                                             save_pb_at_end = config.get("save_pb", 0)
                                             ))

        # summary hook
        if config["save_summaries"]:
            save_steps_summaries = self._get_steps(config["save_summaries_period"], self._time_reference_str)

            self.set_summaries()

            summary_hooks = [tf.train.SummarySaverHook(save_steps=save_steps_summaries,
                                                       output_dir=self._tensorboard_dir+sn.name,
                                                       summary_op=sn,
                                                       summary_writer=fw)
                             for sk in self.summary_keys for sn,fw in zip(self.summary_nodes[sk], self.summary_writers[sk])]

            hooks += summary_hooks

        # images input hook
