-
我的PyTorch Lightning项目代码片段集合：pytorch-lightning-snippets
资源介绍
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