文件名称：pytorch_tempest:我使用pytorch-lightning和hydra训练神经网络的仓库
文件大小：66KB
文件格式：ZIP
更新时间：2024-06-13 16:08:47
deep-learning hydra hacktoberfest training-pipeline pytorch-lightning
暴风雨 该存储库提供了用于训练神经网络的管道。 使用的主要框架： 管道的主要思想： 所有参数和模块都在配置中定义； 事先为不同的优化器/调度器等准备配置，因此很容易在它们之间进行切换； 有用于不同深度学习任务的模板。 当前，支持图像分类和命名实体识别。 运行管道的示例：这将在MNIST上进行训练（将下载数据）： >>> python train.py --config-name mnist_config model.encoder.params.to_one_channel=True 默认运行： >>> python train.py 管道的默认版本在imagenette数据集上运行。 为此，请从此存储库下载数据： : 解压缩并在conf / datamodule / image_classification.yaml路径中定义其路径。