引言

Batch size问题

在Tensorflow-GNN中使用batch size除了需要注意上面的链接问题之外,最近我在调试的发现,使用了merge_batch_to_components() 之后,使用TF-GNN的Readout模块,它会默认merge之后的graph为一张图读出所有节点的数据组成一个矩阵,而不区分batch中的每一张子图,故会导致数据的结构被修改,导致模型的表现与预期的差距较大。

解决方案

out = tfgnn.keras.layers.Pool(tfgnn.CONTEXT, "sum", node_set_name="node")(graph)

使用Pool替代Readout,该代码的具体的作用是从merge之后的图中,读出每一张子图(component)上 的节点数据,并对每个子图的节点数据进行pooling,如这里使用加法做为pooling的方式,并把这些子图pooling之后的数据拼接成一个矩阵存储在context中。这个矩阵的数据和原始输入的graph是一一对应的,如输入的batch size是32,这个矩阵的行数也为32行,每一行对应一张graph。

最新文章

  1. [Java Collection]List分组之简单应用.
  2. C#测试web服务是否可用
  3. AC日记——铺地毯 洛谷 P1003(水水水水水~)
  4. jquery sortTable拖拽排序
  5. 点击其它地方隐藏div/事件冒泡/sweet-alert阻止冒泡
  6. Mysql binlog日志解析
  7. Android textview 设置不同的字体大小和颜色
  8. javaEE的13种核心技术规范
  9. poj3903 Stock Exchange(最长上升子序列)
  10. 卸载php+apache+mysql
  11. freemarker字符串转换成日期和时间(十三)
  12. 【算法导论】单源最短路径之Bellman-Ford算法
  13. js使用面向对象编写下拉菜单
  14. nginx 配置文件的结构
  15. ros新建的包找不到
  16. solrCloud 4.9 分布式集群部署及注意事项
  17. #leetcode刷题之路36-有效的数独
  18. Dynamics CRM plugin调试方法之Profiler
  19. bash's parameter expansion
  20. linux 修改openfiles

热门文章

  1. Graph Theory の brief introduction
  2. 题解【P1833 樱花】
  3. 4. 利用MySQL Shell安装部署MGR集群 | 深入浅出MGR
  4. 【面试题】纯css实现三角形,你知道如何实现吗?
  5. Spring源码 07 IOC refresh方法2
  6. 记一次 ClickHouse 性能测试
  7. Ansible yaml 剧本(傻瓜式)
  8. HDFS的读写流程——宏观与微观
  9. 【NOI P模拟赛】最短路(树形DP,树的直径)
  10. 「SHOI2014」概率充电器