Commit 5fc1e2f9 authored by Yuxin Wu's avatar Yuxin Wu

fix missing arguments in LocalCLIDebugHook (#191)

parent c0afb545
......@@ -84,7 +84,7 @@ class TFLocalCLIDebugHook(HookToCallback):
Refer to tensorflow documentation for details.
"""
from tensorflow.python import debug as tfdbg
super(TFLocalCLIDebugHook, self).__init__(tfdbg.LocalCLIDebugHook())
super(TFLocalCLIDebugHook, self).__init__(tfdbg.LocalCLIDebugHook(*args, **kwargs))
def add_tensor_filter(self, *args, **kwargs):
"""
......
......@@ -512,7 +512,7 @@ class TFDatasetInput(FeedfreeInput):
# TODO theoretically it can support dict
assert isinstance(df, DataFlow), df
assert isinstance(types, (list, tuple)), types
df = MapData(df, lambda dp: tuple(dp))
df = MapData(df, tuple)
df.reset_state()
ds = tf.data.Dataset.from_generator(
df.get_data, tuple(types))
......
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