Skip to content

Fix/embedding input type#36

Open
Alirezak2n wants to merge 2 commits intoyuyangw:masterfrom
Alirezak2n:fix/embedding-input-type
Open

Fix/embedding input type#36
Alirezak2n wants to merge 2 commits intoyuyangw:masterfrom
Alirezak2n:fix/embedding-input-type

Conversation

@Alirezak2n
Copy link
Copy Markdown

Hi, I mostly made changes that shouldn't make any issues in the main model such as removing some imports and formatting/linting. I also had a couple of issues when I wanted to run the finetuning GCN due to missing imports and a missing tensor type.

These changes affect only the style of the code and get rid of redundant imports.
- Cast the first two columns of the node feature tensor (`x`) to `torch.long` for compatibility with embedding layers.
- Addressed the `RuntimeError` caused by incorrect tensor types in the `GCN` model.
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.

1 participant