search

m2lines / gz21_ocean_momentum

Stochastic-Deep Learning Parameterization of Ocean Momentum Forcing
MIT License
5 stars 1 forks source link

Finish neural net refactor at `models/fully_conv_net.py` #108

Open raehik opened 9 months ago

raehik commented 9 months ago

20 / #39 added a module models/fully_conv_net.py refactored from models/models1.py, which holds some basic PyTorch neural net architecture. We never started using it, though-- #43 tracked the work to swap from the old to refactored modules.

We should clean up related code if we're to use it. The "mixin" superclasses and final_transformation code are wonky, and should be simplified.

Supplants #40 , #43 .

raehik commented 9 months ago

See #63 for another related issue.