 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
|
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
|
c07bf05324
|
add PDF2SeQ
|
2025-03-10 19:02:42 +08:00 |
|
czzhangheng
|
0a9ac1a025
|
fix bugs
|
2025-03-03 16:31:04 +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 |