import tensorflow as tf


        self._plot_model_hooks_kwargs = {"time_reference" : self._time_reference_str,
                                         "plot_offset": self._plot_offset}

        self._n_steps_stats = self._get_steps(config["stats_period"], self._time_reference_str)

        # stop hook
        tot_steps = int(self._opts['epochs']+1)*self.n_batches_per_epoch
        hooks.append(tf.train.StopAtStepHook(last_step=tot_steps))

        # general info hook (no average on validation but only on train loop)
        hooks.append(self._create_general_info_hook(config))

        # regularizers hook (no average on validation but only on train loop)
        hooks.append(self._create_regularizers_hook(config))

