Saving loading hook state#400

Open

NaziaHossain066 wants to merge 3 commits into

tgm-team:mainfrom

NaziaHossain066:saving-loading-hook-state

Commits on Apr 11, 2026

Add state_dict/load_state_dict to stateful hooks and checkpoint save/resume to tgat.py
NaziaHossain066
committed
Add unit tests for stateful hook checkpointing and fix checkpoint resume reproducibility
NaziaHossain066
committed
Add --resume/--checkpoint-dir args and save best_val in checkpoints
NaziaHossain066
committed