Add TensorBoard graph for model visualization (#4464)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-21 01:22:29 +02:00
parent 87ce15d383
commit 3acead7e79
6 changed files with 29 additions and 36 deletions
--- a/docs/reference/models/yolo/segment/train.md
+++ b/docs/reference/models/yolo/segment/train.md
@ -12,7 +12,3 @@ keywords: Ultralytics, YOLO, SegmentationTrainer, image segmentation, object det
 ---
 ## ::: ultralytics.models.yolo.segment.train.SegmentationTrainer
 <br><br>
-
---
-## ::: ultralytics.models.yolo.segment.train.train
-<br><br>
--- a/docs/reference/models/yolo/segment/val.md
+++ b/docs/reference/models/yolo/segment/val.md
@ -12,7 +12,3 @@ keywords: Ultralytics, YOLO, SegmentationValidator, model segmentation, image cl
 ---
 ## ::: ultralytics.models.yolo.segment.val.SegmentationValidator
 <br><br>
-
---
-## ::: ultralytics.models.yolo.segment.val.val
-<br><br>
--- a/docs/reference/utils/callbacks/tensorboard.md
+++ b/docs/reference/utils/callbacks/tensorboard.md
@ -13,6 +13,10 @@ keywords: Ultralytics, YOLO, documentation, callback utilities, log_scalars, on_
 ## ::: ultralytics.utils.callbacks.tensorboard._log_scalars
 <br><br>

+---
+## ::: ultralytics.utils.callbacks.tensorboard._log_tensorboard_graph
+<br><br>
+
 ---
 ## ::: ultralytics.utils.callbacks.tensorboard.on_pretrain_routine_start
 <br><br>