Commit Graph

9 Commits

Author SHA1 Message Date
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 dcd4f99f9c fix PDG2SEQ config 2025-03-10 19:05:26 +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
czzhangheng f35bfbf75d fix DSANET 2025-03-03 15:41:14 +08:00
HengZhang 44517219b7 add dl function 2025-03-03 11:34:36 +08:00
HengZhang 26758e761b init 2025-03-02 23:41:12 +08:00