文件名称:pytorch-template:PyTorch深度学习项目变得容易
文件大小:24KB
文件格式:ZIP
更新时间:2024-05-01 12:47:45
Python
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
【文件预览】:
pytorch-template-master
----base()
--------__init__.py(86B)
--------base_trainer.py(6KB)
--------base_data_loader.py(2KB)
--------base_model.py(648B)
----model()
--------loss.py(102B)
--------metric.py(560B)
--------model.py(729B)
----test.py(3KB)
----train.py(3KB)
----trainer()
--------__init__.py(23B)
--------trainer.py(4KB)
----LICENSE(1KB)
----requirements.txt(52B)
----data_loader()
--------data_loaders.py(658B)
----parse_config.py(6KB)
----.gitignore(1KB)
----logger()
--------logger.py(750B)
--------logger_config.json(869B)
--------__init__.py(50B)
--------visualization.py(3KB)
----README.md(13KB)
----utils()
--------util.py(2KB)
--------__init__.py(20B)
----.flake8(88B)
----new_project.py(810B)
----config.json(971B)