pytorch lightning snippets:我的PyTorch Lightning项目的代码片段集合 源码

PyTorch闪电片段 围绕PyTorch Lightning收集的有用工具。 pip install git+https://github.com/awaelchli/pytorch-lightning-snippets 回呼 监控培训数据 :star: 这已移动到! :star: 回调,记录每个传递给training_step方法的张量的直方图。 对于调试和健全性检查预处理管道很有用。 目前支持TensorBoard和WandbLogger。 from monitor import TrainingDataMonitor from pytorch_lightning import Trainer model = YourLightningModule () monitor = TrainingDataMonitor ( row_log_interval = 25 ) trainer = Trai