Skip to content

Commit 9137958

Browse files
authored
Default hook config (aws#338)
1 parent f23e89c commit 9137958

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

smdebug/tensorflow/keras.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -452,6 +452,8 @@ def save_gradients_from_logs(self, gradients):
452452
layer_name = str(v.numpy(), "utf-8")
453453
elif isinstance(v, tf.Variable):
454454
layer_name = v.name
455+
elif isinstance(v, bytes):
456+
layer_name = str(v, "utf-8")
455457
else:
456458
layer_name = v
457459
layer_name = layer_name.split(":")[0]

0 commit comments

Comments
 (0)