Tensorflow在linux中运行的时候进入的一些坑及自己找到的一些办法

入坑

  • 1:Linux 的批处理文件 .sh文件 命令的连接符问题
  • 2:object dection 测试代码中,测试出来的图像不显示
  • 3:在进行对Google官方代码修改的时候,如果进行对学习率 learning_rate 的改变 — 如果只是直接用tf.assign()赋值操作,则会报一下错误: `‘float’ object has no attribute ‘dtype’
  • `tensorflow/core/framework/allocator.cc:124] Allocation of 708083712 exceeds 10% of system memory.

跳坑

  • 1:还没有解决 …
  • 2:在代码的循环最后,加上 plt.show()即可。
  • 3:这是因为如果在初始化学习率 lr 的时候, TF默认用的是 float 类型,所以在 TF 解析的时候会解析成 python 的 float 类型,而不是 TF 的 float 类型,所以会有错误。
    如果把赋值的 lr 修改成 np.arrar(lr)仍然是不行的,错误仍然存在。 仍然会报错。解决方法是:一定要对 lr 先进行初始化为一个变量,然后对 lr 再赋值。具体执行是:先将 lr 用TF 函数初始化一个变量,然后在自己的优化器 optimizer 中将 lr 作为一个参数传进去,之后可以在 session 中先初始化这个变量,进而再运行 train_step 即可。
  • 4:batch_size过大,调小设定的 batch_size 即可
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值