graphstorm icon indicating copy to clipboard operation
graphstorm copied to clipboard

Modify the node classification module of peft_llm example

Open GentleZhu opened this issue 11 months ago • 0 comments

In https://github.com/awslabs/graphstorm/pull/760, we use GPEFT module as the gnn_encoder for link prediction. We still need to modify the node classification code to use GPEFT component and correct its model save function.

GentleZhu avatar Mar 07 '24 19:03 GentleZhu