文件名称:pytorch-lightning-snippets:我的PyTorch Lightning项目的代码片段集合
文件大小:4.85MB
文件格式:ZIP
更新时间:2024-03-28 07:07:29
Python
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
【文件预览】:
pytorch-lightning-snippets-master
----requirements.txt(26B)
----monitor()
--------module_data_monitor.py(4KB)
--------__init__.py(120B)
--------training_data_monitor.py(1018B)
--------data_monitor_base.py(6KB)
----setup.py(656B)
----README.md(8KB)
----.images()
--------peek.gif(4.45MB)
--------training-data-monitor.png(134KB)
--------module-data-monitor.png(316KB)
----verification()
--------batch_norm.py(3KB)
--------batch_gradient.py(8KB)
--------base.py(5KB)
--------__init__.py(0B)
----tests()
--------test_batch_gradient.py(5KB)
--------requirements.txt(11B)
--------__init__.py(0B)
--------test_data_monitor.py(2KB)
--------templates()
--------test_batch_norm.py(2KB)
----.gitignore(2KB)
----checkpoint()
--------code_snapshot.py(3KB)
--------peek.py(3KB)
--------__init__.py(0B)