mittagessen / kraken

OCR engine for all the languages
http://kraken.re
Apache License 2.0
750 stars 131 forks source link

Add tensorboard support #422

Closed colibrisson closed 1 year ago

colibrisson commented 1 year ago

To help monitor training, I suggest adding tensorboard support by providing a pytorch_lightning.loggers.tensorboard object to KrakenTrainer: https://pytorch-lightning.readthedocs.io/en/latest/api/pytorch_lightning.loggers.tensorboard.html#module-pytorch_lightning.loggers.tensorboard. Monitoring could be triggered by a simple --debug argument.

Tensorboard is already a PL dependency: https://github.com/Lightning-AI/lightning/blob/d155ecc7ca4e93ece61a15f5b5f3ba3d45ed3639/requirements/pytorch/test.txt

If you think it could be helpful, I can do it next week.

PonteIneptique commented 1 year ago

I'd love this and something like wandb to avoid ssh ING to check results 😊

On Sun, 29 Jan 2023, 11:16 am Colin Brisson, @.***> wrote:

To help monitor training, I suggest adding tensorboard support by providing a pytorch_lightning.loggers.tensorboard object to KrakenTrainer:

https://pytorch-lightning.readthedocs.io/en/latest/api/pytorch_lightning.loggers.tensorboard.html#module-pytorch_lightning.loggers.tensorboard. Monitoring could be triggered by a simple --debug argument.

If you think it could be helpful, I can do it next week.

— Reply to this email directly, view it on GitHub https://github.com/mittagessen/kraken/issues/422, or unsubscribe https://github.com/notifications/unsubscribe-auth/AAOXEZTU6R7QE4QIVDKFJCDWUY7PRANCNFSM6AAAAAAUKFE6RM . You are receiving this because you are subscribed to this thread.Message ID: @.***>

mittagessen commented 1 year ago

Sure it would be helpful for having a machine-readable log without turning to the API. If you open a pull request reasonably quickly I can merge it for the next stable release.