Commit Graph

9 Commits

Author SHA1 Message Date
zjut
fa8106838e feat(model): 重构模型并添加新功能
- 重新组织了模型结构,增加了新的特征融合模块
- 添加了深度可分离卷积块和新的细节特征提取模块
- 更新了数据处理流程,使用了新的数据集路径
- 调整了训练参数,增加了训练轮次和学习率- 优化了损失函数,使用了Huber损失替代MSE损失
2024-11-12 10:37:56 +08:00
zjut
f4b3a933bf train:调整训练日志输出频率
- 将训练日志的输出频率从每100 个批次调整为每个批次
- 此修改可以提供更频繁的训练进度反馈,便于实时监控训练状态
2024-10-26 19:04:12 +08:00
5e561ab6f7 修改代码结构,提高可读性和可维护性;调整训练输出频率。
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
2024-10-09 11:35:06 +08:00
96ce7d5fda 修改代码结构,提高可读性和可维护性;调整训练输出频率。
改进 self.enhancement_module 为
        self.enhancement_module = WTConv2d(32, 32)
2024-10-08 16:50:11 +08:00
d65f6ec3ce 减少训练过程中的输出频率
修改了训练脚本,将训练过程中的进度输出频率从每批输出改为每100批输出一次,以减少输出量,提高训练效率。
2024-10-06 22:16:42 +08:00
7068b627c4 增加了INN部分的残差连接模块,修改了训练和测试代码以提高代码的可读性和可维护性。- 在train.py中添加了打印所有参数的代码,以方便检查和记录 2024-10-06 16:58:13 +08:00
faacea007c 修改代码实现,提高代码可读性和可维护性 2024-10-06 16:42:18 +08:00
c9e054e236 修改代码实现,提高代码可读性和可维护性 2024-10-05 20:49:35 +08:00
HXY13
da5da74611
Add files via upload 2024-06-03 19:36:29 +08:00