我使用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 {
}
}
}
}
}