-
Notifications
You must be signed in to change notification settings - Fork 9
Open
Labels
enhancementNew feature or requestNew feature or request
Description
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): |
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request