tensorflow
CrystalheartLi
路漫漫其修非常远兮,吾将上下左右前后而求索。
展开
-
tensorflow1.X读取checkpoint(加载训练好的模型及参数)的两种方式
在模型训练的保存路径下面会保存有多个模型(如下图),下面介绍两种读取方式1,获取最新的保存的一个模型,比如上图的model-5200# model_path是checkpoints的路径,下面方法会自动寻找最新保存的一个模型ckpt_file = tf.train.latest_checkpoint(model_path)saver.restore(sess,ckpt_file)2,获取某个迭代次数的模型,比如要指定获取上图的model-5199ckpt_file=os.path.join(原创 2022-03-25 10:57:10 · 2813 阅读 · 2 评论 -
解决‘Hint: If you want to see a list of allocated tensors when OOM happens, add ................‘
tensorflow-gpu运行ner模型的训练代码,出现报错如下图:主要原因是显卡的显存不够。解决方法:1,降低训练的数据输入批次值大小:batch_size2, 将模型放到gpu显存大的服务器上跑Done!!!原创 2021-05-10 14:43:01 · 1102 阅读 · 0 评论 -
tensorflow.python.framework.errors_impl.DataLossError:Unable to open table file xxx perhaps ........
tensorflow加载预训练模型报错:tensorflow.python.framework.errors_impl.DataLossError: Unable to open table file /home/chinese_L-12_H-768_A-12/bert_model.ckpt.data-00000-of-00001: Data loss: not an sstable (bad magic number): perhaps your file is in a different file原创 2021-05-10 13:43:08 · 2434 阅读 · 0 评论 -
解决报错“The name tf.train.Optimizer is deprecated. Please use tf.compat.v1.train.Optimizer instead“
我用tensorflow_gpu2.0以上版本和1.14版本跑一个项目均报这个错误。后来降低到tensorflow_gpu1.5版本,又报错:然后我又把版本降低到1.10,就不出错了。注意tensorflow版本改变后需要看是否cuda还与之匹配,点此查看tensorflow-gpu与cuda等环境对应版本所以,tensorflow_gpu很有可能在1.11,1.12,1.13版本的时候更改了tf.train.Optimizer的接口,具体在哪个版本改得我还不明确,希望有大神看见留言解答。Don原创 2021-05-07 14:01:51 · 4937 阅读 · 0 评论 -
tensorflow_gpu与python、cuda、cudnn、gcc对应版本--官方最新
1,点击进入官方网址查看最新2,截至本文发表之前:Done!!!原创 2021-05-07 13:57:42 · 1801 阅读 · 1 评论 -
测试tensorflow,pytorch,paddlepaddle是否可正常调用GPU
1,tensorflowimport tensorflow as tfprint(tf.test.is_gpu_available())如果输出以下结果说明调用成功:2,pytorchimport torchprint(torch.cuda.is_available())如果输出以下结果说明调用成功:3,paddlepaddleimport paddlepaddle.fluid.install_check.run_check()如果输出以下结果说明调用成功:Running原创 2021-04-14 18:39:11 · 2426 阅读 · 1 评论 -
AttributeError: module ‘tensorflow‘ has no attribute ‘Session‘
使用tensorflow2.X版本执行tf.Session()会报错:AttributeError: module 'tensorflow' has no attribute 'Session'原因:tensorflow2.X以上版本已经移除了Session()模块解决办法:将tf.Session()改为tf.compat.v1.Session()原创 2021-01-10 12:15:43 · 95 阅读 · 0 评论 -
win10系统Pycharm下cuda10.1+cudnn7.5+tensorflow-gpu+python3.7的环境配置
win10系统Pycharm下cuda10.1+cudnn7.5+tensorflow-gpu+python3.7的环境配置花了整整一天的时间终于搞明白了,并测试了一下(如下图),最近很忙,这里先不具体写了,过几天有时间了再更。留个微信:429470804,着急的朋友可以加微信问我,一定尽全力帮大家解决问题。或者大家可以先看看这篇文章(https://blog.csdn.net/lixc316/...原创 2019-05-11 05:48:34 · 1043 阅读 · 0 评论