Pytorch-Template是一款用于生成pytorch模板文件的工具

PyTorch模板项目 pytorch模板文件生成器，它支持数据集，数据加载器，模型，优化器，损失，优化器和lr_scheduler的多处理程序。 要求 Bash（Linux） Python> = 3.6 requirements.txt 特征 清晰的文件夹结构，适用于许多深度学习项目。 .json配置文件支持，方便进行参数调整。 支持多数据集，多数据加载器，多模型，多优化器，多损耗，多优化器和多lr_scheduler。 以上所有内容都可以在.json配置中构建！ 通过在.bashrc中添加PATH，您可以在所有路径下执行torch_new_project.py 。 可自定义的命令行选项可更方便地进行参数调整。 检查点保存和恢复。 抽象基类，可加快开发速度： BaseTrainer处理检查点的保存/恢复，训练过程日志记录以及初始化各种对象。 BaseDataset处理