Commit Graph

233 Commits

Author SHA1 Message Date
7c85b2b8cb chore: 完善验证流程 2026-05-13 21:05:44 +08:00
5f542fb577 style: 调整代码风格问题 2026-05-13 18:25:05 +08:00
5d95027894 refactor: 重写沙比 AI 写的训练代码 2026-05-13 18:20:31 +08:00
6746e96994 chore: 定义依赖版本 2026-05-12 22:38:36 +08:00
08b90881a8 fix: 三阶段一起训练
Co-authored-by: Copilot <copilot@github.com>
2026-05-07 22:12:33 +08:00
3676958781 feat: 分三阶段训练
Co-authored-by: Copilot <copilot@github.com>
2026-05-07 20:59:22 +08:00
3b86a78e0b feat: 预训练加入前一阶段的信息
Co-authored-by: Copilot <copilot@github.com>
2026-05-06 22:51:57 +08:00
7535ecc9fe chore: 调整损失值计算 2026-05-06 22:47:08 +08:00
f0025df1ec fix: focal loss
Co-authored-by: Copilot <copilot@github.com>
2026-05-06 22:27:56 +08:00
3ed3ad8238 feat: 优化 focal loss
Co-authored-by: Copilot <copilot@github.com>
2026-05-06 21:55:34 +08:00
b9032d94c8 refactor: 数据处理也将资源统一
Co-authored-by: Copilot <copilot@github.com>
2026-05-06 21:37:30 +08:00
ea57bbde3a chore: 修改超参数 2026-05-06 20:44:00 +08:00
d4f0768d2f fix: 二三阶段训练
Co-authored-by: Copilot <copilot@github.com>
2026-05-05 23:50:01 +08:00
b8f691269d fix: masked_focal
Co-authored-by: Copilot <copilot@github.com>
2026-05-05 22:09:53 +08:00
850c038be3 chore: 修改超参数 2026-05-05 21:49:44 +08:00
3e273c5a9d feat: 资源压缩为一种 tile
Co-authored-by: Copilot <copilot@github.com>
2026-05-05 21:46:38 +08:00
3874d4dd95 feat: VQ 编码器分为三部分
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 18:41:57 +08:00
dcb85dc1f8 chore: 调整超参数 2026-04-28 17:51:39 +08:00
3e962bf8be chore: 调整训练 sh 脚本 2026-04-28 17:51:02 +08:00
6460d8c5bc feat: 损失值改为 focal loss
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 17:23:51 +08:00
f22943820c feat: 提高 VQ 解码头容量
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 16:50:15 +08:00
294e214431 chore: 调整超参数 2026-04-28 16:46:02 +08:00
a69403d6bf feat: vq 预训练
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 16:31:53 +08:00
789107969b fix: MaskGIT 输出地图不更新编码器
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 15:53:41 +08:00
6b2b36f20c chore: 调整超参数 2026-04-28 15:34:36 +08:00
b34007a691 feat: z 一致性约束
Co-authored-by: Copilot <copilot@github.com>
2026-04-28 15:32:37 +08:00
fed377ae31 fix: 堆叠报错
Co-authored-by: Copilot <copilot@github.com>
2026-04-27 16:07:36 +08:00
81d49944cc feat: 随机验证分为有条件和无条件
Co-authored-by: Copilot <copilot@github.com>
2026-04-27 15:36:11 +08:00
9bddb05625 feat: 图片输出添加标签说明 2026-04-27 15:31:50 +08:00
638c107884 chore: 参数调整 2026-04-27 15:20:28 +08:00
abbad781ab feat: 添加少数结构性标签
Co-authored-by: Copilot <copilot@github.com>
2026-04-27 14:56:21 +08:00
21315a6cb0 chore: 调整超参数与细节
Co-authored-by: Copilot <copilot@github.com>
2026-04-27 13:46:48 +08:00
cc463f543e fix: 数据集路径 2026-04-26 23:49:17 +08:00
068940cae0 refactor: 采用 VQ + MaskGIT 方案
Co-authored-by: Copilot <copilot@github.com>
2026-04-26 23:45:56 +08:00
1eda704986 refactor: heatmap 模型采用预测 x_0 而非噪声
Co-authored-by: Copilot <copilot@github.com>
2026-04-25 17:09:48 +08:00
b471bb46eb fix: Diffusion 训练热力图改为 -1-1 范围 2026-04-25 16:18:16 +08:00
966d007721 chore: 添加 epoch 之间的检查 2026-04-25 15:28:29 +08:00
3be014f3ad chore: 调整损失值计算 2026-04-23 19:24:17 +08:00
164bb24823 feat: 联合训练预览
Co-authored-by: Copilot <copilot@github.com>
2026-04-23 18:37:10 +08:00
90cfe54bd2 fix: 联合训练报错 2026-04-23 18:24:18 +08:00
765cdcaeb0 feat: 冻结 MaskGIT 并联合训练 2026-04-22 22:15:05 +08:00
4059f0e05a chore: 调整训练 2026-04-10 12:49:21 +08:00
54164b9f22 chore: 换回第一版 diffusion 策略 2026-04-10 12:46:34 +08:00
1237d45d95 chore: 调整参数 2026-04-10 12:41:53 +08:00
df23c891c6 fix: MaskGIT 训练调整 2026-04-09 12:47:18 +08:00
9814edd1b4 fix: 数据集报错 2026-04-08 19:42:18 +08:00
dbb0b9064c feat: 降低 MaskGIT 对条件的依赖 2026-04-08 19:36:46 +08:00
cbbe312444 fix: 报错 2026-04-08 13:04:27 +08:00
5df14a3f3f fix: heatmap 验证索引 2026-04-07 22:57:52 +08:00
3e898dc5ba feat: 调整 Diffusion 模型 2026-04-07 22:56:41 +08:00