Pytorch-lightning: logger: add TensorBoard without tensorFlow

Created on 5 Dec 2019  ·  7Comments  ·  Source: PyTorchLightning/pytorch-lightning

Is your feature request related to a problem.
any way to get rid of the TensorFlow dependency?

Describe alternatives you've considered
See discussion in #265.

Additional context

enhancement help wanted

Most helpful comment

I think I have a little time today to take a crack at it.

All 7 comments

Found some additional info.
The FAQ in the tensorboard readme states:

Can I run tensorboard without a TensorFlow installation?
TensorBoard 1.14+ can be run with a reduced feature set if you do not have TensorFlow installed. The primary limitation is that as of 1.14, only the following plugins are supported: scalars, custom scalars, image, audio, graph, projector (partial), distributions, histograms, text, PR curves, mesh. In addition, there is no support for log directories on Google Cloud Storage.

As far as I know, PyTorch provides the SummaryWriter but does not have TensorBoard/TensorFlow as a requirement.

@awaelchli nice, are you interested in submitting a PR? :]

yes! this is a big addition. @awaelchli or @neggert would anyone be interested in adding this today so we can have it in the next release?

would be interested. However, I have unfortunately no time today. @neggert feel free to take it if you want and have the time.

here are clear instructions on how to do this:

https://pytorch-lightning.readthedocs.io/en/latest/Trainer/Logging/#custom-logger

I think I have a little time today to take a crack at it.

Was this page helpful?
0 / 5 - 0 ratings