File size: 392 Bytes
0b32ad6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from pathlib import Path
try:
# HACK: SummaryWriter must be imported at the begining or else it will lead to core dumped
# This is a known issue: https://github.com/pytorch/pytorch/issues/30651
from torch.utils.tensorboard.writer import SummaryWriter
except:
pass
with (Path(__file__).parent.resolve() / "version.txt").open() as file:
__version__ = file.read().strip()
|