Question

当我使用keras VGG16预训练模型在tf.dataset.map函数中获得瓶颈功能时，我遇到了这个问题

vgg16 = VGG16(include_top=False, input_shape=input_shape)
vgg16.trainable = False


def _parse_function(example_proto):
  features = {'img': tf.FixedLenFeature((), tf.string, default_value=""),
              'label': tf.FixedLenFeature((), tf.int64, default_value=0)}
  parsed_features = tf.parse_single_example(example_proto, features)
  img = tf.decode_raw(parsed_features['img'], tf.uint8)
  img = tf.reshape(img, (224, 224, 3))
  img = tf.cast(img, tf.float32) / 255.0
  img = tf.expand_dims(img, 0)
  img = vgg16(img)
  img = tf.squeeze(img, [0])
  label = parsed_features['label']
  return img, label

ds.map(_parse_function)

错误打印是

2019-06-19 13：47：19.378807：E tensorflow / core / common_runtime / executor.cc：624]执行程序无法创建内核。无效的参数：默认MaxPoolingOp仅在设备类型CPU上支持NHWC [[{{node vgg16 / block1_pool / MaxPool}}]] 2019-06-19 13：47：19.379204：W tensorflow / core / framework / op_kernel.cc：1401] OP_REQUIRES在iterator_ops.cc:660失败：无效参数：默认MaxPoolingOp仅在设备类型CPU上支持NHWC [[{{node vgg16 / block1_pool / MaxPool}}]] 追溯（最近一次通话）： _do_call中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，行1334 返回fn（* args） _run_fn中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，第1319行选项，feed_dict，fetch_list，target_list，run_metadata） _call_tf_sessionrun中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，行1407 run_metadata） tensorflow.python.framework.errors_impl.InvalidArgumentError：默认MaxPoolingOp仅在设备类型CPU上支持NHWC [[{{node vgg16 / block1_pool / MaxPool}}]] [[{{node MakeIterator}}]]

在处理上述异常期间，发生了另一个异常：

回溯（最近通话最近）：在第103行的“ /home/tf/image_cls/train.py”文件中 steps_per_epoch = steps_per_epoch，适合的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/engine/training.py”，行776 shuffle = shuffle） _standardize_user_data中第2200行的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/engine/training.py” K.get_session（）。run（x.initializer）运行中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，第929行 run_metadata_ptr） _run中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，行1152 feed_dict_tensor，选项，run_metadata） _do_run中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，行1328 run_metadata） _do_call中的文件“ /opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/client/session.py”，行1348 提高类型（e）（node_def，op，消息） tensorflow.python.framework.errors_impl.InvalidArgumentError：默认MaxPoolingOp仅在设备类型CPU上支持NHWC [[{{node vgg16 / block1_pool / MaxPool}}]] [[node MakeIterator（在/home/tf/image_cls/train.py:103处定义）]]

tensorflow.python.framework.errors_impl.InvalidArgumentError：默认MaxPoolingOp仅在设备类型CPU上支持NHWC

0 个答案: