Enable tensorboard visualization

salesforce / LAVIS

LAVIS - A One-stop Library for Language-Vision Intelligence

BSD 3-Clause "New" or "Revised" License

9.98k stars 974 forks source link

Open StanLei52 opened 2 years ago

StanLei52 commented 2 years ago

Hi, thank you for the great work!

I wonder if there is any plan to incorporate the tensorboard visualization. Also if there is any plan to combine pytorch_lightning?

dxli94 commented 2 years ago

Hi @StanLei52, thanks for your interest.

We are working on improving the logging, including tensorboard visualization.

We don’t have an immediate plan to support lightning yet. We may add a separate lightning runner depending on how intrusive the changes required are.

thanks!

suhmily commented 1 year ago

May I ask how to use tensorboard visualization in Lavis?