Commit 6b2bc083 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Log encoder scaffold arguments.

PiperOrigin-RevId: 327518640
parent 0fddab49
...@@ -22,6 +22,7 @@ from __future__ import print_function ...@@ -22,6 +22,7 @@ from __future__ import print_function
import inspect import inspect
from absl import logging
import gin import gin
import tensorflow as tf import tensorflow as tf
...@@ -205,6 +206,8 @@ class EncoderScaffold(tf.keras.Model): ...@@ -205,6 +206,8 @@ class EncoderScaffold(tf.keras.Model):
super(EncoderScaffold, self).__init__( super(EncoderScaffold, self).__init__(
inputs=inputs, outputs=outputs, **kwargs) inputs=inputs, outputs=outputs, **kwargs)
logging.info('EncoderScaffold configs: %s', self.get_config())
def get_config(self): def get_config(self):
config_dict = { config_dict = {
'num_hidden_instances': self._num_hidden_instances, 'num_hidden_instances': self._num_hidden_instances,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment