Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit c65a646

Browse files
Lukasz KaiserRyan Sepassi
authored andcommitted
Disable image summary in eager mode (tf.Eager doesn't like it for now).
PiperOrigin-RevId: 177309768
1 parent 7909c69 commit c65a646

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensor2tensor/layers/modalities.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,8 @@ class ImageModality(modality.Modality):
205205
def bottom(self, inputs):
206206
with tf.variable_scope(self.name):
207207
inputs = common_layers.standardize_images(inputs)
208-
tf.summary.image("inputs", inputs, max_outputs=2)
208+
if not self._model_hparams.use_eager_mode:
209+
tf.summary.image("inputs", inputs, max_outputs=2)
209210
return tf.to_float(inputs)
210211

211212
def targets_bottom(self, inputs):

0 commit comments

Comments
 (0)