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

feat: add tensorboard support #206

Merged
merged 4 commits into from
Aug 27, 2024
Merged

feat: add tensorboard support #206

merged 4 commits into from
Aug 27, 2024

Conversation

Franklalalala
Copy link
Contributor

@Franklalalala Franklalalala commented Aug 22, 2024

Add TensorBoard Support. And it has the following features:

  • add a new parameter in train_options section, use_tensorboard.
  • Set true to use tensorboard. It will record iteration error once every 25 iterations, epoch error once per epoch.
  • There are tree types of error will be recorded. train_loss_iter is iteration loss, train_loss_last is the error of the last iteration in an epoch, train_loss_mean is the mean error of all iterations in an epoch.
  • Learning rates are tracked as well.
  • A folder named tensorboard_logs will be created in the working directory. Use tensorboard --logdir=tensorboard_logs to view the logs.

Fix #207

@QG-phy
Copy link
Collaborator

QG-phy commented Aug 26, 2024

@Franklalalala 这个单元测试没过。你修一下代码~应该应该加一个安装库的依赖就好。另外,上传一个例子吧。

@QG-phy
Copy link
Collaborator

QG-phy commented Aug 26, 2024

安装的问题,我已经解决了。 @Franklalalala 请你上传一个例子吧。另外这个PR的描述(你看最上面)我修改了一下留下了XXX的待补充,你补充说明一下这个PR支持了什么,好吧。

@QG-phy QG-phy self-requested a review August 26, 2024 09:16
@QG-phy QG-phy added the Feature Request New feature or request label Aug 26, 2024
@Franklalalala
Copy link
Contributor Author

添加了测试案例在 example 里, argcheck 做了相应的修改,可以控制是否添加 tensorboard,默认是否,不添加

@QG-phy QG-phy self-requested a review August 27, 2024 09:46
@QG-phy QG-phy merged commit ac6faa9 into deepmodeling:main Aug 27, 2024
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Feature Request New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add TensorBoard Support
2 participants