Skip to content

TensorBoard plugin blocks TensorBoard UI from fully loading #338

@psybuzz

Description

@psybuzz

Repro:

  • In a fresh Python virtualenv, pip install tensorboard torch-tb-profiler
  • Open TensorBoard (http://localhost:6006 by default)
  • The UI under the header is blank, but the loading spinner rotates
    image
  • In the DevTools network panel, the '/plugins_listing' request is stalled
  • Eventually showing an error
    image

On my machine, I tested this with

  • tensorboard==2.5.0
  • torch-tb-profiler==0.2.0

From a brief read, torch-tb-plugin's is_active() method seemingly issues blocking call to wait for data:

def is_active(self):
"""Returns whether there is relevant data for the plugin to process.
"""
self._is_active_initialized_event.wait()
return self._is_active

On the TensorBoard side, there have been a few reports from users who find TensorBoard responsive again after uninstalling torch-tb-plugin: tensorflow/tensorboard#5076, tensorflow/tensorboard#5111

TensorBoard's other builtin plugins are implemented with is_actives that either return quickly or are non-blocking. Would it be possible for torch-tb-plugin to do something similar?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions