The current DenseNet implementation makes the axis parameter of BatchNormalization function 1. However, this parameter should be decided by the data format, thus channel first or channel last.
I fixed this error by querying the data format first. Then pass the concatenate axis as a parameter to the function.
tdeboissiere
commented
6 years ago
The current DenseNet implementation makes the axis parameter of BatchNormalization function 1. However, this parameter should be decided by the data format, thus channel first or channel last.
I fixed this error by querying the data format first. Then pass the concatenate axis as a parameter to the function.