Deeponet

[ba2tro]

Resolves #37. I have added the files for DeepONet implementation from OperatorLearning.jl and added tests for the architecture.

[ba2tro]

@foldfelis should I update the readme and docs as well?

[foldfelis]

Hi @Abhishek-1Bhatt it would be great to have the docs and readme updated 😊

[ba2tro]

Great 😊, I'll start working on it.

[codecov]

Codecov Report

Merging #39 (d3e58f1) into master (f47637f) will decrease coverage by 8.10%.
The diff coverage is 71.42%.

@@             Coverage Diff             @@
##            master      #39      +/-   ##
===========================================
- Coverage   100.00%   91.89%   -8.11%     
===========================================
  Files            2        5       +3     
  Lines           52       74      +22     
===========================================
+ Hits            52       68      +16     
- Misses           0        6       +6     

Impacted Files	Coverage Δ
src/NeuralOperators.jl	100.00% <ø> (ø)
src/DeepONet.jl	60.00% <60.00%> (ø)
src/subnets.jl	100.00% <100.00%> (ø)

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update f47637f...d3e58f1. Read the comment docs.

[foldfelis]

Hi @Abhishek-1Bhatt I didn't notice that there is a model training task inside the test 🤣 . I was wondering why the CI kept running for an hour. Is there another way to test DeepONet?

[ba2tro]

Oh😅, this was intended to be a regression test, just to sanity check that our model's training does not get deteriorated if we try out something new in the future. Indeed it doesn't make sense to train for 400 epochs here, I see that now : ) . One thing we can do is to reduce the no. of epochs to 100 or even lesser and then check against the error bound or would you like to have a different way of testing for it? @ChrisRackauckas might also have some suggestions regarding regression testing here.

[foldfelis]

Usually, I only do the unit test and put the training tasks into the example folder. I will manually test the examples locally to make sure that things won't go wrong. I am looking for a more efficient way to test the robustness of the models as well.

[ba2tro]

That seems like a nice way to do it, we can transfer it to example/Burgers

[foldfelis]

Feel free to add new examples and I will reorganize and refactor the project once the new featur is merged.

[ba2tro]

Removed an import of MAT.jl, it was causing the tests to fail😅

[foldfelis]

Looks good to me 🎉