Different style loss

[SmitSheth]

According to paper https://arxiv.org/pdf/1508.06576.pdf (Page 11) the style defined is different that the one that you have used. Any reason for using different one ?

[Permafacture]

Do you mean that the code should use the dimensions of the style filter bank at that layer (style.shape) for N and M (from equation 4 in the paper you linked) instead of channels and size*width? Or are you referring to something else?