pytorch

1
import tensorboard
2
from torch._vendor.packaging.version import Version
3

4
if not hasattr(tensorboard, "__version__") or Version(
5
    tensorboard.__version__
6
) < Version("1.15"):
7
    raise ImportError("TensorBoard logging requires TensorBoard version 1.15 or above")
8

9
del Version
10
del tensorboard
11

12
from .writer import FileWriter, SummaryWriter  # noqa: F401
13
from tensorboard.summary.writer.record_writer import RecordWriter  # noqa: F401
14