Problem occurring at the projection concatenation.

[TParcollet]

Dear all,

I feel like you have made a mistake (or let a test value on the code :) ) on at the line 128 of training.py, indeed I think that the concatenation should be on the same axis as the two previous one so channel_axis and not this value. When I'm using 1 such as in the code I get a weird tensor shape like (32,16,44) while I should have (16,16,44).

Hope that will help !