czzhangheng
/
Project-I
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Secret Projct
21 Commits 4 Branches 0 Tags 116 MiB
Python 100%
Go to file
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
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模型添加node到edge的Linear转换层 2025-09-11 11:42:02 +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