Project-I/models
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
..
STDEN 为STDEN模型添加node到edge的Linear转换层 2025-09-11 11:42:02 +08:00
STGODE 更新STGODE模型:添加配置文件、优化模型代码、新增测试数据文件 2025-09-03 10:18:52 +08:00
STGODE_LLM 添加STGODE_LLM和STGODE_LLM_GPT2模型实现,更新配置文件和README 2025-09-04 11:12:11 +08:00
STGODE_LLM_GPT2 优化READMER,注册STGODE-LLM 2025-09-04 06:05:09 +00:00
model_selector.py 优化READMER,注册STGODE-LLM 2025-09-04 06:05:09 +00:00