czzhangheng
/
Project-I
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
21 Commits 4 Branches 0 Tags 116 MiB
Commit Graph

2 Commits

Author SHA1 Message Date
czzhangheng 626bb4d2bb 为STDEN模型添加node到edge的Linear转换层 
- 在STDENModel中添加node_to_edge和edge_to_node转换层
- 修改forward方法以处理node_num输入并输出node_num格式
- 更新编码器以处理edge格式的中间数据
- 修复解码器中的维度计算问题
- 解决设备不匹配和数据类型不一致问题
- 更新.gitignore以允许models/STDEN/代码目录被跟踪

现在模型可以接受node_num格式的输入，内部转换为edge_num进行处理，最后转换回node_num输出。
 2025-09-11 11:42:02 +08:00
czzhangheng ab5811425d add STDEN 2025-09-03 08:11:01 +08:00