On bn_final in cnn_learner/create_cnn

fastai
1

Hi all,

in function cnn_learner/create_cnn, there is an argument bn_final, with default False. What is the effect for normal image classification tasks if we set it to True?

Anyone has any idea?
Thanks.

2

It creates a latch BatchNorm layer for the output. We have found it sometimes help, so don’t hesitate to try it for yourself!

2 Likes
3

Did you mean add a BatchNorm layer at the last?

4

Yes, just before the output.