使用NasNet模型与keras做深度学习训练时采用以下代码
一、代码片:
inputs = Input((224, 224, 3))
base_model = NASNetMobile(include_top=False, input_shape=(224, 224, 3)) # , weights=None
x = base_model(inputs)
刚开始一直报这样的错误:
ValueError: padding must be zero for average_exc_pad Apply node that
caused the error: AveragePoolGrad{ignore_border=True,
mode=‘average_exc_pad’, ndim=2}(Elemwise{Composite{(i0 * (i1 +
Abs(i1)))}}.0, IncSubtensor{InplaceInc;::, ::, :int64:, :int64:}.0,
TensorConstant{(2,) of 2}, TensorConstant{(2,) of 2},
TensorConstant{(2,) of 1}) Toposort index: 137 Inputs types:
[TensorType(float32, 4D), TensorType(float32, 4D), TensorType(int32,
vector), TensorType(int32, vector), TensorType(int32, vector)] Inputs
shapes: [(32, 32, 64, 64), (32, 32, 33, 33), (2,), (2,), (2,)] Inputs
strides: [(524288, 1