1. 程式人生 > >Tensorflow 錯誤記錄

Tensorflow 錯誤記錄

Tensorflow 程式碼出錯問題記錄

  1. TypeError: unsupported operand type(s) for *: ‘int’ and ‘Dimension’
    修改:將出現錯誤地方的使用了整數的地方新增int();
    原因:使用了Tensor的shape直接取使用,這是不可以的,應當改為int(x.shape[1])這樣。
  2. TypeError: Value passed to parameter ‘input’ has DataType float64 not in list of allowed values: float16, float32
    修改:使用tf.cast(x, dtype=tf.float32) 將資料型別統一。
  3. 如果出現OOM的錯誤警告:一種情況是:GPU正在執行別的程式;另外一種情況是:訓練的引數過大,或者batchsize過大導致的。