Skip to content

Add WSMTHP model and WSM-specific T handling#79

Merged
iLampard merged 1 commit intoant-research:mainfrom
KenCao2007:add-wsmthp
Apr 10, 2026
Merged

Add WSMTHP model and WSM-specific T handling#79
iLampard merged 1 commit intoant-research:mainfrom
KenCao2007:add-wsmthp

Conversation

@KenCao2007
Copy link
Copy Markdown
Contributor

What

This PR adds a new model, WSMTHP, to EasyTPP.

Changes

  • add easy_tpp/model/torch_model/torch_wsm_thp.py
  • register WSMTHP in easy_tpp/model/__init__.py
  • add config examples in examples/configs/experiment_config.yaml
  • add WSM-specific observation-window T handling support

Notes

  • existing EasyTPP model interfaces are unchanged
  • the added T logic is only used by WSMTHP
  • existing models are unaffected

@iLampard
Copy link
Copy Markdown
Collaborator

Hi, will you update readme, to introduce this model with source, as well?

@iLampard iLampard merged commit c4338ed into ant-research:main Apr 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants