netrans/bin/torch/utils/tensorboard/__init__.py

7 lines
339 B
Python

try:
from tensorboard.summary.writer.record_writer import RecordWriter # noqa F401
except ImportError:
raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '
'This should be available in 1.14 or above.')
from .writer import FileWriter, SummaryWriter # noqa F401