因为创建InceptionV3模型是正确的吗?

时间:2018-06-10 19:37:25

标签: tensorflow keras

我尝试按如下方式创建InceptionV3我的第一个模型:

inc_model=InceptionV3(include_top=False, weights='imagenet', input_shape=(3, 150, 150))

所有内容都适用于CPU版本,但当我尝试使用CUDA版本时,我收到以下错误:

  

ValueError:Shape必须为1级,但排名为4   ' batch_normalization_1 / cond / F usedBatchNorm' (op:' FusedBatchNorm')   输入形状:[?,32,149,1],[1,32,1,1],[1,32,1,1],[1,32,1,1],   [1,32,1,1]。

0 个答案:

没有答案