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

Add tensorboard #209

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

Add tensorboard #209

wants to merge 6 commits into from

Conversation

florian-huber
Copy link
Contributor

@florian-huber florian-huber commented Mar 6, 2024

So far only a simply writer for use with Tensorboard.

  • tracks loss and rmse for train and val set
  • could be expanded: e.g. by adding writer.add_graph to also track the model architecture

@florian-huber florian-huber marked this pull request as ready for review March 6, 2024 13:13
Copy link
Collaborator

@niekdejonge niekdejonge left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I have not worked with tensorboard before so it is a bit hard to judge for me if this is implemented correctly. What I understand from the code is that some logging data will be stored in the folder "runs", which I assume can later be used to visualize part of the model in tensorboard.

Base automatically changed from dev_pytorch to main April 3, 2024 10:32
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.

2 participants