【文件属性】:
文件名称:lightning-kitti:使用 Pytorch-Lightning 进行语义分割
文件大小:243KB
文件格式:ZIP
更新时间:2021-05-30 10:55:46
JupyterNotebook
闪电猫
使用 Pytorch-Lightning 进行语义分割
介绍
这是一个简单的演示,用于使用对执行语义分割,并通过使用监控和比较运行来优化神经网络。
Pytorch-Lightthning 包括一个 W&B 记录器,可以简单地调用:
from pytorch_lightning . loggers import WandbLogger
from pytorch_lightning import Trainer
wandb_logger = WandbLogger ()
trainer = Trainer ( logger = wandb_logger )
有关更多详细信息,请参阅。
超参数可以手动定义,每次运行都会自动记录到以便更轻松地分析/解释结果以及如何优化架构。
您还可以运行以自动优化超参数。
注意:本示例改编自 Pytorch-Lightning 示例。
用法
【文件预览】:
lightning-kitti-master
----models()
--------unet()
----Pipfile(204B)
----train.py(8KB)
----sweep.yaml(544B)
----notebook()
--------tutorial.ipynb(189KB)
----imgs()
--------results.png(47KB)
--------sweep.png(176KB)
----requirements.txt(41B)
----.gitignore(51B)
----README.md(3KB)