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
  1. tf.function是不允许修改input的,graph.cache是一个dict类型的input,如果在tf.function中计算缓存(而不是在之前计算),会导致cache被修改(插入了计算的归一化矩阵)
  2. 两层GCN的归一化设置是一样的,所以用第一层去计算缓存的话,第二层也是可以用到的