PyTorch Lightning
www.pytorchlightning.ai › blog › tensorboard-with-pyLightning gives us the provision to return logs after every forward pass of a batch, which allows TensorBoard to automatically make plots. We can log data per batch from the functions training_step (),validation_step () and test_step (). We return a batch_dictionary python dictionary.
GitHub - pcmin03/lightning-template
github.com › pcmin03 › lightning-templateIntroduction. This template tries to be as general as possible. It integrates many different MLOps tools. Effective usage of this template requires learning of a couple of technologies: PyTorch, PyTorch Lightning and Hydra. Knowledge of some experiment logging framework like Weights&Biases, Neptune or MLFlow is also recommended.