models/classifier/l2_norm code error

[bhpugongying]

Hi. I think the weight normalization should be along the dim of 1 since the size of weight is (2, 512). But in the code, the weight normalization is along the dim of 0.

[goldiusleonard]

There's nothing wrong with it during my training process. Maybe you cna try it out again