Question Related the Norm Term in Linear Layer

[Eric-Gty]

Hi there, really thanks for your amazing work. When reading your source code, I found one place that I don't really understand its meaning. In your regression_nf.py (model), you re-construct the Linear layer and add a norm term. Later, when you regress the out_coord, you use the norm term; and when you regress the sigma, you didn't use that.

Thus, I would like to know what is that norm term for? Why do you use it for the coordinate regression?

Hope to get some feedback from you, thanks :))