Secret Projct

Go to file

harry.zhang c1a4962795 README FIX		2025-09-04 06:11:50 +00:00
.idea	README FIX	2025-09-04 06:11:50 +00: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	优化READMER，注册STGODE-LLM	2025-09-04 06:05:09 +00:00
trainer	更新项目文件：添加数据选择器、模型选择器、训练器选择器和ODE训练器	2025-09-03 07:52:18 +08:00
utils	update_README	2025-09-04 03:53:02 +00:00
.gitignore	Remove PEMS08 from tracking and update .gitignore to ignore PEMS08/PEMSD8 datasets	2025-09-04 06:09:17 +00: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	update_README	2025-09-04 03:53:02 +00: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