Currently, the log_graph function in the tensorboard writer only supports the tuple type for the sample input. It should at least be extended to support a dictionary input as well, which is fairly common with HF style inputs.
|
if isinstance(inputs, tuple): |
Currently, the log_graph function in the tensorboard writer only supports the tuple type for the sample input. It should at least be extended to support a dictionary input as well, which is fairly common with HF style inputs.
PyMarlin/pymarlin/utils/writer/tensorboard.py
Line 102 in fb8bbc9