-
PyTorch-template使PyTorch深度学习项目实现简化
资源介绍
PyTorch模板项目
PyTorch深度学习项目变得容易。
指标
附加记录
验证数据
检查点
张量板可视化
贡献
待办事项
执照
致谢
要求
Python> = 3.5（建议3.6）
PyTorch> = 0.4（建议1.2）
tqdm（对于test.py可选的）
tensorboard> = 1.14（请参见Tensorboard可视化）
特征
清晰的文件夹结构，适用于许多深度学习项目。
.json配置文件支持，方便进行参数调整。
可自定义的命令行选项，可更方便地进行参数调整。
检查点保存和恢复。
抽象基类，可加快开发速度：
BaseTrainer处理检查点的保存/恢复，培训过程记录等。
BaseDataLoader处理批处理生成，数据改组和验证数据拆分。
BaseModel提供基本的模型摘要。
资料夹结构
pytorch-template/
│
├── tr