Skip to content

Conversation

@florian-huber
Copy link
Member

@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.

3 participants