- 在STDENModel中添加node_to_edge和edge_to_node转换层 - 修改forward方法以处理node_num输入并输出node_num格式 - 更新编码器以处理edge格式的中间数据 - 修复解码器中的维度计算问题 - 解决设备不匹配和数据类型不一致问题 - 更新.gitignore以允许models/STDEN/代码目录被跟踪 现在模型可以接受node_num格式的输入,内部转换为edge_num进行处理,最后转换回node_num输出。 |
||
|---|---|---|
| .. | ||
| STDEN_modules.md | ||
| diffeq_solver.py | ||
| ode_func.py | ||
| stden_model.py | ||
| utils.py | ||