Commit Graph

30 Commits

Author SHA1 Message Date
czzhangheng c15cf605be 添加md结果 2025-04-26 15:37:40 +08:00
czzhangheng 5f8c31af2e 添加STIDGCN 2025-04-23 23:24:43 +08:00
czzhangheng e826240a5e Merge remote-tracking branch 'origin/main'
# Conflicts:
#	model/model_selector.py
2025-04-23 23:24:05 +08:00
czzhangheng 97eb39073a 添加STIDGCN 2025-04-23 23:22:50 +08:00
czzhangheng bc9a2667c2 新增了模型蒸馏STMLP
现在Trainer每次epoch完后都会保存模型checkpoint
其中STMLP会自动教师模型到pre-train
根据教师模型的存在情况启动/预训练or蒸馏模式
2025-04-07 17:05:59 +08:00
czzhangheng 229b6320b9 添加固定种子 2025-04-04 14:37:05 +08:00
czzhangheng e8fc67b867 简化NLT, DDGCRN, GWN代码 2025-03-27 21:28:39 +08:00
czzhangheng 1b76cc6ce2 优化GWN, ddgcrn的代码,使其更简洁易读 2025-03-27 20:08:17 +08:00
czzhangheng d016dd5980 新增消耗分析模式,只需在原有的mode中调整为benchmark即可 2025-03-27 20:07:26 +08:00
czzhangheng 8c839642e1 新增消耗分析模式,只需在原有的mode中调整为benchmark即可 2025-03-27 20:02:16 +08:00
czzhangheng 5306d24408 新增消耗分析模式,只需在原有的mode中调整为benchmark即可 2025-03-26 12:38:57 +08:00
czzhangheng dcd4f99f9c fix PDG2SEQ config 2025-03-10 19:05:26 +08:00
czzhangheng ae7e91d162 Merge remote-tracking branch 'origin/main' 2025-03-10 19:02:52 +08:00
czzhangheng c07bf05324 add PDF2SeQ 2025-03-10 19:02:42 +08:00
HengZhang 7679507523 add url 2025-03-03 16:45:34 +08:00
czzhangheng 0a9ac1a025 fix bugs 2025-03-03 16:31:04 +08:00
czzhangheng f35bfbf75d fix DSANET 2025-03-03 15:41:14 +08:00
czzhangheng 18aaa32c10 更新 transfer_guide.md 2025-03-03 15:25:51 +08:00
HengZhang 9181413d09 Update transfer_guide.md 2025-03-03 15:22:05 +08:00
HengZhang 01782a0c81 update Readme Add transfer_guide.md 2025-03-03 13:42:08 +08:00
HengZhang 5df1c810b1 edit README 2025-03-03 13:19:03 +08:00
HengZhang d2b0ac5f9a Merge branch 'main' of https://github.zhang-heng.com/czzhangheng/TrafficWheel 2025-03-03 13:01:21 +08:00
HengZhang 2e2b5608ae add expResult 2025-03-03 12:59:34 +08:00
czzhangheng d2a4a65dc6 develop pip 2025-03-03 11:54:23 +08:00
czzhangheng 085c4c2514 update pip 2025-03-03 11:52:09 +08:00
HengZhang 2bb113fed6 Update README.md 2025-03-03 11:38:32 +08:00
HengZhang 44517219b7 add dl function 2025-03-03 11:34:36 +08:00
HengZhang be5e810c54 mv dir name 2025-03-03 10:29:42 +08:00
HengZhang 26758e761b init 2025-03-02 23:41:12 +08:00
czzhangheng 3f77b2759b Initial commit 2025-03-02 23:39:13 +08:00