Deep-Learning-Project-Template

PyTorch 项目模板 一个简单且设计合理的结构对于任何 深度学习 项目都是至关重要的,因此,经过大量实践和在 pytorch 项目中贡献之后,我们提供了一个 pytorch 项目模板,该模板结合了简单性、最佳文件结构实践和良好的面向对象设计。 其主要思想是,每次开始一个 pytorch 项目时,你都会做很多相同的事情,因此,将这些共享内容封装起来,可以帮助你在开始新的 pytorch 项目时只需要更改核心思想。

为了减少重复的工作,我们建议使用高级库。您可以编写自己的高级库,或者只需使用一些第三方库,如 ignite、fastai、mmcv 等。这可以帮助您在几行代码中编写紧凑但功能齐全的训练循环。下面我们使用 ignite 来训练 mnist 作为示例。

目录表

• 概要
• 详细信息
• 未来工作
• 贡献
• 致谢

概要

概括来说，以下是如何使用此模板的，因此例如假设您想实现 ResNet-18 来训练 mnist，那么您应该执行以下操作：

• 在 modeling 文件夹中创建一个名为 Python 的文件，这里我们将其命名为 example_model.py。在 modeling/__init__.py 文件中，您可以创建一个名为 build_model 的函数，以调用您的模型