__init__.py 440 B

12345678910111213
  1. import tensorboard
  2. from distutils.version import LooseVersion
  3. if not hasattr(tensorboard, "__version__") or LooseVersion(
  4. tensorboard.__version__
  5. ) < LooseVersion("1.15"):
  6. raise ImportError("TensorBoard logging requires TensorBoard version 1.15 or above")
  7. del LooseVersion
  8. del tensorboard
  9. from .writer import FileWriter, SummaryWriter # noqa: F401
  10. from tensorboard.summary.writer.record_writer import RecordWriter # noqa: F401