fix(callbacks/lightning): populate callback for `lightning`

XuehaiPan / nvitop

An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.

https://nvitop.readthedocs.io

Apache License 2.0

4.56k stars 144 forks source link

Closed XuehaiPan closed 8 months ago

XuehaiPan commented 8 months ago

Populate the implementation for GpuStatsLogger callback to nvitop.lightning based on lightning.pytorch..callbacks.Callback.

Fixes #112 Closes #113