log_graph is pretty restrictive

[shatu]

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

if isinstance(inputs, tuple):