How to customize the logger in Trainer for detailed graphs

[cemox35]

Hello there,

I want to loss/epoch, acc graphs, etc. However, I only see the tensorboard loss/step and val_loss/step. How can I customize/add it?

Thank you.

[mpariente]

By subclassing System and using the logging logic of lightning (using self.log).

[cemox35]

I used the predefined Conv-Tasnet model. Also, I am beginner. Can you show some code examples? Thank you for your help.

[mpariente]

Have a look at the docs of pytorch-lightning, and try to make a subclass of System for logging. If you show me your attempt, I'll correct it.