- 博客(3)
- 收藏
- 关注
原创 更换优化器出现OOM问题
请问一下,在调用Bert optimization中自定义的优化器时出现OOM,更换成tf.train.AdamOptimizer就OK了,是什么原因?代码如下:with tf.variable_scope("optimizer"): num_warmup_steps = int(num_train_steps * config['warmup_proportion']) # 导入bert自定义optimization self.train_op = optimization.c
2020-10-21 09:47:21
474
原创 module 'tensorflow.contrib' has no attribute 'linalg'
module ‘tensorflow.contrib’ has no attribute ‘linalg’在tensorflow 1.13版本中,运行tf.contrib.linalg.LinearOperatorTriL(diag_vals).to_dense()报错:tf.contrib.linalg.LinearOperatorTriL(diag_vals).to_dense()主要是...
2019-12-04 20:06:14
1415
3
原创 Transformer学习总结附TF2.0代码实现
Transformer学习总结 TF2.0代码实现Transformer1.Transformer理论详解1.1 transformer总体架构1.2 输入部分如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能,丰富你的文章UML ...
2019-12-01 21:06:24
6010
14
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人