2018年11月21日
tensorflow 在加载大型的embedding模型参数时，会遇到cannot be larger than 2GB
摘要： 这种问题是，对于每一个变量 variable 由于是基于protobuf存在这大小限制(2G),这个时候，我们需要将embedding拆开，拆分成N等分，来使得每一个 variable都在2G以下; 结果为: 阅读全文
posted @ 2018-11-21 16:35 龚细军 阅读(2310) 评论(1) 推荐(0) 编辑