Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TensorBoard add_graph() fix #3236

Merged
merged 1 commit into from
May 19, 2021
Merged

TensorBoard add_graph() fix #3236

merged 1 commit into from
May 19, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 19, 2021

Possible fix for #3235

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved TensorBoard graph logging in YOLOv5 training routine.

πŸ“Š Key Changes

  • Uncommented code for TensorBoard graph logging.
  • Adjusted the position of TensorBoard image logging comment (code itself is not changed; just the comment position).

🎯 Purpose & Impact

  • πŸš€ Purpose: The uncommented lines enable the visualization of the model's computational graph in TensorBoard, which can help in debugging and understanding the model architecture.
  • πŸ›  Impact: Users who utilize TensorBoard for monitoring and analyzing model performance will now have easier access to visual representations of the model's graph without having to modify the code themselves. However, this could also mean slightly increased resource usage during training due to graph logging.

@glenn-jocher glenn-jocher linked an issue May 19, 2021 that may be closed by this pull request
@glenn-jocher
Copy link
Member Author

glenn-jocher commented May 19, 2021

This PR will run the TensorBoard add_graph() method for all training runs now, so model architecture is viewable in the 'GRAPHS' section.

Screenshot 2021-05-19 at 13 49 29

@glenn-jocher glenn-jocher merged commit b7cd1f5 into master May 19, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-2 branch May 19, 2021 11:51
Lechtr pushed a commit to Lechtr/yolov5 that referenced this pull request Jul 20, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Is there any picture to discribe the architecture of yolov5?
1 participant