How to manually log to Ray Train's internal Tensorboard logger?

263 Views Asked by At

Ray Train automatically stores various things to Tensorboard. In addition, I want to log custom histograms, images, PR curves, scalars, etc. How do I access Ray Train's internal TBXLogger so that I can log additional things?

1

There are 1 best solutions below

4
On

Hi Ray team member here. What ray train version are you using? I don't think we currently have an end to end example to showcase how to do this. But you should be able to have something like tf.summary.image in your training function. Note this would log images per training worker.