Meeting an error

The program is leaving me an error while running
For "cnn = Conv2D(64, (3, 3), activation='relu')(input_image)" in line 41
An error returns like
"ValueError: The channel dimension of the inputs should be defined. Found None."

Keras version == 2.1.5

Thanks for your contribution!~

[bojone]

check your ~/.keras/keras.json.
the image_data_format must be channels_last

[RYFan-RS]

I used you model on a 224*224 RGB dataset ,and get a error of "loss = nan" or both loss and acc don't changes .

[bojone]

@FlyDogFan

"loss not change" seems like capsule's problem, not my implement~

you can try to replace K.epsilon() with 1e-3 in squash function in Capsule_Keras.py to solve nan problem.

[liuyang1273]

I used you model on a 60*80 RGB dataset ,and get a error of acc don't changes . @bojone

[Ponyo1]

Thank you. After change the squash function. The model work properly.