Logging hooks: Changes from TensorboardX to tensorboard.

pesser / edflow

Reduce boilerplate code for your ML projects. TensorFlow and PyTorch.

https://edflow.readthedocs.io/

MIT License

24 stars 13 forks source link

Logging hooks: Changes from TensorboardX to tensorboard. #239

Closed ffeldmann closed 4 years ago

ffeldmann commented 4 years ago

This PR changes the current usage from tensorboardX to tensorboard. Since PyTorch natively supports tensorboard it would be nice to use it instead of tensorboardX.

Added Features: Added log_tensorboard_figures functionality to add matplotlib figures to the logging mechanism.

ffeldmann commented 4 years ago

Please review and /or approve.

ffeldmann commented 4 years ago

Review plz.