import tensorflow as tf

    def _init_session(self, sess, model):
        w = self._train_params['image_size']
        h = self._train_params['image_size']
        in_ch = 3
        m = model

        # Do initialization of all variables
        sess.run(tf.global_variables_initializer())

        # Load datasets with defaults
        sess.run([m.train_dataset_init_op, m.pred_dataset_init_op], feed_dict={
            m.ph.train_images: np.zeros((1, w, h, in_ch)),
            m.ph.train_classes: np.zeros((1,)),
            m.ph.pred_images: np.zeros((1, w, h, in_ch)),
