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

VIP专享 2021-03-08 13:20:49 4.85MB ZIP
91
收藏 收藏
举报

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

...展开详情
立即下载 身份认证后 购VIP低至7折
一个资源只可评论一次,评论内容不能少于5个字
您会向同学/朋友/同事推荐我们的CSDN下载吗?
谢谢参与!您的真实评价是我们改进的动力~
  • 至尊王者

关注 私信
上传资源赚钱or赚积分
最新推荐
pytorch-lightning-snippets:我的PyTorch Lightning项目的代码片段集合-源码 (VIP专享) VIP下载
1/0