Finish some TODOs in comments

m2lines / gz21_ocean_momentum

Stochastic-Deep Learning Parameterization of Ocean Momentum Forcing

MIT License

5 stars 1 forks source link

Closed tztsai closed 1 year ago

tztsai commented 1 year ago

Added a decorator in analysis/analysis.py to allow a function to specify a save_file parameter that saves the plot.
Implement some TODOs in models/base.py.
Added a --device argument in the CMD argument parser to specify the device used for training.