frugally-deep: Invalid axis for batch normalization

Loading the model crashes on “Invalid axis for batch normalization”. According to the code, the else statement is “Just to make the compiler happy. In reality, this is never called.” Well, in my case it is… See output below. I’m attaching a link to my .h5 and .json files. Please tell me what I’m doing wrong. https://drive.google.com/file/d/1afQKjJjnWEg_gv4-eKiHUQpnIz1XRU_m

Loading json ... done. elapsed time: 3.645327 s
Building model ... done. elapsed time: 61.173442 s
Running test 1 of 1 ... terminate called after throwing an instance of 'std::runtime_error'
  what():  Invalid axis for batch normalization.
Aborted (core dumped)