-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Feature: Interfacing ML code with MiMA #10
base: master
Are you sure you want to change the base?
Conversation
Update to reflect model now takes 3 inputs and concatenates them, rather than a pre-concatted input. Test is very basic but shows that TF davenet does the same thing as PyTorch davenet.
It was there to test something.
The models' arguments were being sent in the wrong order. It should be pressure, latitude, then wind. Also the pressure should be in Pa, so no need for scale factor of 100.
Closes #13 in the early @SimonClifford commits. |
@SimonClifford commits address TensorFlow coupling. |
@jatkinson1000 commits close #11 and close #12 and therefore close #4 . |
@jatkinson1000 closes #5, although we may revisit. |
Also closes #1 in earliest @jatkinson1000 commits (we could merge CMake docs are included in |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A few small suggestions.
- Move the CSD3 submissions scrips into a
scripts\csd3
folder as they are hardware specific. - Remove hardcoded home directories in the submissions scripts. Replace with a variable or argument.
- pytorch README.md could do with a bit more detail on what the scripts are. This could be opened as an issue for future work though.
Closes #7
@jatkinson1000 - covering #4 and #5
cg_drag.f90
to accommodate drag forcing from various different approachesThis way each approach will call the NN twice, once with zonal, once with meridional, but tensors will only be written to shared memory once.
@SimonClifford - covering #3
We then need to conduct benchmarking: