Viewing File: /home/ubuntu/.local/lib/python3.10/site-packages/tensorboardX/__init__.py

"""A module for visualization with tensorboard
"""

from .record_writer import RecordWriter
from .torchvis import TorchVis
from .writer import FileWriter, SummaryWriter