nbren12 / uwnet

Neural Networks based unified physics parameterization for atmospheric models
MIT License
3 stars 2 forks source link

Refactor loss functions and logging #38

Closed nbren12 closed 5 years ago

nbren12 commented 5 years ago

This commit refactors equilibrium penalty into its own function, computes the maximum and minimum WTG eigenfuctions. It stores all these loss functions in a dictionary. This dictionary is printed for each iteration.