Можно ли запустить «With tf.Session as sess:» внутри training_data_generator при запуске fit_generator?

Question

Я использую tf.Session в тензорной доске, поэтому не могу включить режим Eager.

Мне нужно извлечь патчи изображения в большом изображении, используя tf.image.extract_image_patches. Таким образом, в моем собственном учебном генераторе я добавляю что-то вроде:

While True:
     num_patches = tf.image.extract_image_patches(input_big_pic, ksizes, strides, rates, patch_padding)
     With tf.Session as sess:
          inputs_after_tensor1 = sess.run(num_patches )
     .....some modifies for this ndarray "inputs_after_tensor1"....... 
     yield ({'input1': np.array(result_inputs_after_tensor1)})

Моя функция потерь уменьшается, но мои выходные изображения в тензорном потоке не меняются, поэтому мне интересно, влияет ли tf.Session в моем training_data_generator на fit_generator?

Answer 1

Я пытался, и я думаю, что можно запустить tf.session внутри, но это замедляет процедуру обучения.