Commit Graph

12 Commits

Author SHA1 Message Date
czzhangheng a3e43fc6df 添加支持REPST In D8 2025-11-09 16:27:32 +08:00
czzhangheng 221b67aa53 标准化PEP8格式 2025-11-08 21:26:03 +08:00
czzhangheng 04f65856f6 更新 .gitignore 2025-08-19 00:01:00 +08:00
harry.zhang 8025a46baa 更新.gitignore以忽略Result.xlsx文件,修改run.py以优先使用macOS的MPS设备,优化设备设置逻辑 2025-08-18 14:32:20 +08:00
czzhangheng bef30b9c2f 解决合并冲突,整合dev和main分支的更改 2025-05-14 13:13:11 +08:00
czzhangheng eb8684bf91 收到垃圾堆 2025-04-22 14:54:39 +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
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