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
import inspect
from absl import logging
import gin
import tensorflow as tf
......@@ -205,6 +206,8 @@ class EncoderScaffold(tf.keras.Model):
super(EncoderScaffold, self).__init__(
inputs=inputs, outputs=outputs, **kwargs)
logging.info('EncoderScaffold configs: %s', self.get_config())
def get_config(self):
config_dict = {
'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