我尝试按如下方式创建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]。