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

这个tf_utils的加速，为什么一定要在第一层的gcn使用build_cache_for_graph呢？ #41

Open

jiadiwu opened this issue Apr 14, 2022 · 1 comment

jiadiwu commented Apr 14, 2022

No description provided.

Member

hujunxianligong commented Apr 14, 2022

tf.function是不允许修改input的，graph.cache是一个dict类型的input，如果在tf.function中计算缓存（而不是在之前计算），会导致cache被修改（插入了计算的归一化矩阵）
两层GCN的归一化设置是一样的，所以用第一层去计算缓存的话，第二层也是可以用到的

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