tensorflow报错问题
主要是在使用tensorflow中遇到的问题记录
chenmingwei000
这个作者很懒,什么都没留下…
展开
-
tfrecord 预测是batch_size的维度为?号原因
由于在于预测是,每一个数据都不能丢弃,所以在tfrecord的读取时tf.contrib.data.map_and_batch( lambda record: _decode_record(record, name_to_features), batch_size=batch_size, drop_remainder=drop_remainder))drop_remainder的设置True Fal...原创 2021-05-12 15:12:17 · 290 阅读 · 0 评论 -
关于bidirectional_dynamic_rnn出现 Dimensions of inputs should match问题
在搭建双向BIRNN模型的时候,调用tensorflow自动展开函数bidirectional_dynamic_rnn(cell_fw, cell_bw, data, dtype=tf.float32)时候出现异常:InvalidArgumentError (see above for traceback): ConcatOp : Dimensions of inputs should match...原创 2020-03-16 16:42:43 · 1017 阅读 · 0 评论 -
针对tensorflow 出现nan的问题解决,transformer训练出现nan的问题总结
先说明一下我运用的是韩国大佬https://github.com/Kyubyong/transformer.git的代码,来做了一个闲聊的问答模 情况说明: 由于在之前一直在用tensortflow-gpu ,在之前服务器通过问答数据处理好之后,运行这个模型,没有报错,并且运行成功,效果好不错。 更换公司后,服务器发生变化,但是环境仍然一致,都是pyt...原创 2019-08-16 16:35:32 · 4783 阅读 · 1 评论 -
安装cudn的一些笔记
https://blog.csdn.net/Edwards_June/article/details/79067318上面的网址是用来禁用已有的显卡驱动;https://www.nvidia.cn/Download/index.aspx这个是驱动对应你的电脑版本寻找网址https://developer.nvidia.com/cuda-gpus哪些支持cudnlspci | g...原创 2018-08-08 16:30:50 · 1296 阅读 · 0 评论