czzhangheng
/
Project-I
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Secret Projct
22 Commits 4 Branches 0 Tags 116 MiB
Python 100%
Go to file
czzhangheng abdd3165b8 修复STDEN模型bug：参数量异常和维度错误 
问题分析：
1. 参数量异常小(16,522) - 缺少node到edge转换层
2. 维度错误 - 编码器期望edge格式但收到node格式输入
3. 解码器维度计算错误

修复内容：
- 添加node_to_edge和edge_to_node转换层，参数量从16,522增加到1,009,002
- 修改forward方法正确处理node格式输入输出
- 修复编码器以处理edge格式的中间数据
- 修正解码器中的维度计算问题

测试结果：
- 参数量：1,009,002 (合理范围)
- 输入输出形状正确：(batch_size, seq_len/horizon, num_nodes, input/output_dim)
- 模型可以正常前向传播
 		2025-09-11 12:39:46 +08:00
configs 添加STGODE_LLM和STGODE_LLM_GPT2模型实现，更新配置文件和README 2025-09-04 11:12:11 +08:00
data Remove PEMS08 from tracking and update .gitignore to ignore PEMS08/PEMSD8 datasets 2025-09-04 06:09:17 +00:00
models 修复STDEN模型bug：参数量异常和维度错误 2025-09-11 12:39:46 +08:00
trainer 更新项目文件：添加数据选择器、模型选择器、训练器选择器和ODE训练器 2025-09-03 07:52:18 +08:00
utils update_README 2025-09-04 03:53:02 +00:00
.gitignore 为STDEN模型添加node到edge的Linear转换层 2025-09-11 11:42:02 +08:00
LICENSE Initial commit 2025-09-01 11:03:24 +08:00
README.md README FIX 2025-09-04 06:11:50 +00:00
main.py 更新项目文件：添加数据选择器、模型选择器、训练器选择器和ODE训练器 2025-09-03 07:52:18 +08:00
requirements.txt 合并远程更改并解决冲突 2025-09-11 11:16:34 +08:00

README.md

Project-I

Secret Projct

mkdir -p models/gpt2

Prepare Env.

pip install -r requirement.txt

Download dataset

python utils/download.py

Download gpt weight

mkdir -p models/gpt2

Download config.json & pytorch_model.bin from https://huggingface.co/openai-community/gpt2/tree/main

wget https://huggingface.co/openai-community/gpt2/resolve/main/config.json?download=true -O ./models/gpt2/config.json
wget https://huggingface.co/openai-community/gpt2/resolve/main/pytorch_model.bin?download=true -O ./models/gpt2/config.json

Use pytorch >= 2.6 to load model.

Run

python main.py --config configs/STGODE_LLM_GPT2/PEMS08.yaml