Skip to content
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

Implement CNODE training #3

Merged
merged 3 commits into from
Mar 5, 2024
Merged

Implement CNODE training #3

merged 3 commits into from
Mar 5, 2024

Conversation

SCiarella
Copy link
Member

I have implemented the second part of Example2, so now it is possible to train the CNODE to learn the Gray-Scott model.

I have opened a PR because @luisaforozco started looking at the current main branch that contains some features that have been removed. In particular, to implement the ML training I had to get rid of the following:

  • BlockMatrices objects: they are too memory-hungry
  • Lanczos4opencv interpolation: this method is not differentiable, so I replaced it with pooling and upsampling from Lux.

Please @luisaforozco merge the PR if you don't have any local conflict!


Disclaimer: the ML is implemented but is extremely slow on my laptop, so I can not confirm yet that it is going to learn the correct thing.

This was referenced Mar 4, 2024
@luisaforozco
Copy link
Contributor

Thanks @SCiarella ! I was also working on this on my side. I will take a look and also push my branch so that we can compare.

@luisaforozco
Copy link
Contributor

I have the impression that the examples are very complex, with many functions doing various things. Could we maybe keep it a bit simpler? Additionally, I also find some of the Julia syntax rather complex, but this might be only me.

I would also say that in the examples, all the functions should be contained in the example notebook/script/md such that you have a clear picture of all elements an can play with them.

@SCiarella SCiarella merged commit 3e9398c into main Mar 5, 2024
@SCiarella
Copy link
Member Author

This branch has been merged to main. The following PR will restructure the examples addressing #7 and #6

@PabRod PabRod deleted the ML_training branch March 12, 2024 12:39
v1kko pushed a commit that referenced this pull request Jan 16, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants