tensorflow相關bug合集 解決方法

2021-09-23 07:49:49 字數 727 閱讀 1166

**報錯回查遇到的一些坑,真的是坑,oom的錯竟然檢查了一天才檢查出來問題所在,故記錄一下遇到的坑。

oom error

解決方案:

顯示卡真的oom了,檢查顯示卡有沒有其他任務;

檢查**;

**無誤的情況下,這種錯90%的概率是因為作處理的資料超過gpu的視訊記憶體,應該減少batch的大小;

merge_all

報錯invalidargumenterror (see above for traceback): you must feed a value for placeholder tensor 『placeholder_19』 with dtype float and shape [?,16,16,17]

[[node placeholder_19 (defined at ***.py:35) ]]

[[}]]

**

self.summary_op = tf.summary.merge_all()
情況出現原因

程式中定義了多個圖

解決方案

tf.summary.scalar('loss', distill_policyloss)

self.summary_op = tf.summary.merge(tf.get_collection(tf.graphkeys.summaries,'loss')) #這裡的loss可以換成其他的op

continued

tensorflow相關函式 to

函式宣告 to categorical y,num classes none,dtype float32 作用 將整型標籤轉為onehot。y為int陣列,num classes為標籤類別總數,大於max y 標籤從0開始的 返回 如果num classes none,返回len y max y 1...

Tensorflow 安裝 使用 相關

最近在ubuntu中折騰tensorflow 花了兩晚上,終於搞定 1.安裝ubuntu 64位,64位,tensorflow是不支援32位的 2.安裝完後,安裝python 2.7 3.安裝pip 再upgrade 一下 4.安裝tensorflow 不用按照中文 那個pip 一大堆,直接pip ...

tensorflow相關tensor計算函式

1.tf.split 該函式主要用於對tensor進行分割,一般在設定多gpu平行計算時經常會被用到,主要是將乙個batch資料集進行平分,分配給各個gpu,最後再彙總各個gpu得到的損失,從而加快模型的訓練速度,其主要引數的定義如下 import tensorflow as tf tf.split...