4. 系统配置
文件config.py用于配置深度学习项目的各种参数和功能,包括数据加载、网络设置、日志记录、目录管理、学习率调整等。文件config.py的具体功能如下:
- 数据配置:通过data_config函数创建数据加载器,支持训练和测试数据集的不同处理。
- 网络配置:network_config函数用于构建模型网络,支持从预训练模型恢复或加载已有模型参数,设置优化器和随机种子。
- 日志配置:log_config函数用于设置日志文件,记录训练过程中的信息。
- 目录配置:dir_config函数检查数据集和标注文件的路径,并创建必要的文件夹来保存日志和模型检查点。