ultralytics 8.0.140 improved robustness to model path spaces (#3879)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Lukas Hennies <45569834+Gornoka@users.noreply.github.com>
2023-07-22 19:44:43 +02:00
parent ed25db9426
commit 965e405957
9 changed files with 85 additions and 25 deletions
--- a/ultralytics/utils/callbacks/tensorboard.py
+++ b/ultralytics/utils/callbacks/tensorboard.py
@ -6,7 +6,9 @@ try:
    from torch.utils.tensorboard import SummaryWriter

    assert not TESTS_RUNNING  # do not log pytest
-except (ImportError, AssertionError):
+
+# TypeError for handling 'Descriptors cannot not be created directly.' protobuf errors in Windows
+except (ImportError, AssertionError, TypeError):
    SummaryWriter = None

 writer = None  # TensorBoard SummaryWriter instance