search

Lightning-Universe / lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
https://lightning-transformers.readthedocs.io
Apache License 2.0
610 stars 77 forks source link

Custom Logging #178

Closed ghomasHudson closed 3 years ago

ghomasHudson commented 3 years ago

🚀 Feature

This may already exist, but it would be great to be able to write plugins for different experiment logging methods, e.g. wandb (weights and biases), tensorboard, etc...

Motivation

For many ML experiments, something more organized than log files is useful, especially for metrics such as the loss, which are best graphed.

Pitch

Would love to be able to add something like python train.py log=wandb, (though maybe more options are needed so log.type="wandb" is better)

SeanNaren commented 3 years ago

I think this has been added in #177 :)

ghomasHudson commented 3 years ago

I think this has been added in #177 :)

Amazing! This project is moving super fast now.