-
Notifications
You must be signed in to change notification settings - Fork 25
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
main.py: error: unrecognized arguments: --agg concat #12
Comments
According to modules/LightGCN.py line 157-166 def pooling(self, embeddings):
# [-1, n_hops, channel]
if self.pool == 'mean':
return embeddings.mean(dim=1)
elif self.pool == 'sum':
return embeddings.sum(dim=1)
elif self.pool == 'concat':
return embeddings.view(embeddings.shape[0], -1)
else: # final
return embeddings[:, -1, :] And utils/parser.py line 33 parser.add_argument("--pool", type=str, default='concat', help="[concat, mean, sum, final]") the "--agg" argument should be "--pool"
fix that typo, issue should be solved. |
Thanks a lot, that completely solves my problem. |
Hi, I run the recommended code
python main.py --dataset ali --gnn ngcf --dim 64 --lr 0.0001 --batch_size 1024 --gpu_id 0 --context_hops 3 --agg concat --ns mixgcf --K 1 --n_negs 64
And received the issue:
main.py: error: unrecognized arguments: --agg concat;
I checked the file /utils/parser.py, and found the --agg parameter is not declared in that file.
Is there anything I missed?
Thanks for your time and have a nice day.
The text was updated successfully, but these errors were encountered: