issues
search
CrawlScript
/
tf_geometric
Efficient and Friendly Graph Neural Network Library for TensorFlow 1.x and 2.x
GNU General Public License v3.0
497
stars
91
forks
source link
这个tf_utils的加速,为什么一定要在第一层的gcn使用build_cache_for_graph呢?
#41
Open
jiadiwu
opened
2 years ago
hujunxianligong
commented
2 years ago
tf.function是不允许修改input的,graph.cache是一个dict类型的input,如果在tf.function中计算缓存(而不是在之前计算),会导致cache被修改(插入了计算的归一化矩阵)
两层GCN的归一化设置是一样的,所以用第一层去计算缓存的话,第二层也是可以用到的