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.

Sign up for GitHub

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

Jump to bottom

关于MTGNN #21

Open

teethdocker opened this issue Feb 25, 2024 · 1 comment

teethdocker commented Feb 25, 2024

您好，我有问题想向您请教，我看到文中实验加入了MTGNN的比较，有比较好的效果，但我对MTGNN源代码进行改写后进行实验出现了OOM问题，请问您是如何处理这个问题的呢？

Collaborator

YunhaoZhang-Mars commented Feb 26, 2024

在模型结构上减小hidden state的数量，训练时用较小的batch size；另外还需要将原实现中的loss改为MSE loss。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment