【分布式Tensorflow(0.11.0)问题未解决】 alexnet_v2/pool1/MaxPool : tensor_in must be 4-dimensional

最新推荐文章于 2020-08-06 01:25:51 发布

__Sunny__

最新推荐文章于 2020-08-06 01:25:51 发布

阅读量2.1k

点赞数

分类专栏： TensorFlow 文章标签： TensorFlow

本文链接：https://blog.csdn.net/s_sunnyy/article/details/71077583

版权

在CentOS 7.1系统上，使用TensorFlow 0.11.0进行分布式训练时遇到了问题。在AlexNet_v2的pool1/MaxPool层，报错提示输入tensor_in必须是4维。尝试通过whl文件安装TensorFlow，但该问题仍未解决。已经在GitHub上发起讨论，目前尚未收到回复。

摘要由CSDN通过智能技术生成

系统：linux Centos 7.1

Tensorflow版本：0.11.0 whl 安装

出错信息：

tensorflow.python.framework.errors_impl.InvalidArgumentError: tensor_in must be 4-dimensional
         [[Node: alexnet_v2/pool1/MaxPool = MaxPool[T=DT_FLOAT, data_format="NHWC", ksize=[1, 3, 3, 1], padding="VALID", strides=[1, 2, 2, 1], _device="/job:worker/replica:0/task:0/cpu:0"](alexnet_v2/conv1/Relu)]]

源代码

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import argparse
from datetime import datetime
import math
import sys
import time

from six.moves import xrange  # pylint: disable=redefined-builtin
import tensorflow as tf

tf.app.flags.DEFINE_string("ps_hosts", "", "Comma-separated list of hostname:port pairs")
tf.app.flags.DEFINE_string("worker_hosts", "", "Comma-separated list of hostname:port pairs")

tf.app.flags.DEFINE_string("job_name", "", "One of 'ps', 'worker'")
tf.app.flags.DEFIN