如何使用由tensorflow训练的模型删除“ cast”节点?

时间:2018-07-17 10:03:42

标签: tensorflow casting

我使用tf.slim训练预训练模型。

“预训练”模型中不包括“ cast”节点,但是使用train_image_classifier.py进行训练后,“ cast”节点会出现。

我使用inception_resnet_v2模型。

这样的节点

node {

  name: "parallel_read/filenames/ToFloat"

  op: "Cast"

  input: "parallel_read/filenames/filenames_Size"

  device: "/device:CPU:0"

  attr {

    key: "DstT"

    value {

      type: DT_FLOAT

    }
  }
  attr {

    key: "SrcT"

    value {
      type: DT_INT32
    }
  }

attr {
     key: "_output_shapes"
     value {
      list {
        shape {
        }
      }
    }
  }
}

0 个答案:

没有答案