-
Notifications
You must be signed in to change notification settings - Fork 3.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
G Retriever Experiments and Improvements (full) #9666
base: master
Are you sure you want to change the base?
G Retriever Experiments and Improvements (full) #9666
Conversation
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
Co-authored-by: Akihiro Nitta <[email protected]>
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
Aiming to resolve the merge conflicts by EoW. I can also break this PR down further if needed. |
# Conflicts: # CHANGELOG.md # test/nn/models/test_g_retriever.py # torch_geometric/datasets/__init__.py # torch_geometric/datasets/web_qsp_dataset.py # torch_geometric/nn/models/g_retriever.py # torch_geometric/nn/nlp/llm.py # torch_geometric/nn/nlp/sentence_transformer.py
for more information, see https://pre-commit.ci
Merge Conflicts are resolved. Currently running the following tests to make sure nothing has broken:
|
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
…iments # Conflicts: # examples/llm_plus_gnn/README.md # examples/llm_plus_gnn/g_retriever.py
for more information, see https://pre-commit.ci
hallucinates_df.to_csv(root_dir + "/hallucinates_df.csv", index=False) | ||
|
||
|
||
def minimal_demo(gnn_llm_eval_outs, dataset, lr, epochs, batch_size, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@puririshi98 FYI: I re-added the minimal demo to this PR, but didn't include it in the critical path of running this file, not sure whether you were going to re-add this minimal demo in a separate PR.
Follow up to PR 9597. Includes multiple changes related to LLM+GNN experiments and scaling up to a remote backend. Including: