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

Separable-PINN in DeepXDE #1776

Draft
wants to merge 34 commits into
base: master
Choose a base branch
from
Draft
Changes from 1 commit
Commits
Show all changes
34 commits
Select commit Hold shift + click to select a range
55663e3
Add JAX backend support for VariableValue callback
bonneted Mar 27, 2024
3b141cc
fix copy paste
bonneted Mar 27, 2024
ab626f7
Merge branch 'lululxvi:master' into jax-variableValue
bonneted Apr 16, 2024
cb562bc
create array wrapper for variable value
bonneted Apr 17, 2024
c51eac6
restore blank line
bonneted Apr 17, 2024
ccbb394
refactor SPINN with list handler
bonneted Apr 18, 2024
e49df54
Add SPINN module to jax/__init__.py
bonneted Apr 18, 2024
c177d86
Add concat and stack functions to jax/tensor.py
bonneted Apr 18, 2024
c639977
create full class for spinn
bonneted Apr 18, 2024
e25523a
fix dimension handling
bonneted Apr 18, 2024
7a16482
Refactor input and output transform handling in NN class
bonneted Apr 18, 2024
51c78de
handle list input for testing
bonneted Apr 18, 2024
d27dcf8
refactor decorator
bonneted Apr 18, 2024
1906448
implement helmot spinn
bonneted Apr 18, 2024
2646d35
Add support for JAX backend in Helmoltz_Dirichlet_2d.py
bonneted Apr 19, 2024
f86641a
Refactor input and output transform handling
bonneted Apr 19, 2024
59a4cc7
reshape instead of squezze to keep one dim
bonneted Apr 22, 2024
a48ba25
dimension handling
bonneted Apr 22, 2024
d62f604
add spinn sampling for hypercube
bonneted Apr 30, 2024
149d7b9
refactor variable wrapper
bonneted Apr 30, 2024
8340f25
remove empty line
bonneted May 2, 2024
3d64359
remove empty line
bonneted May 3, 2024
63de873
fix line
bonneted May 3, 2024
de5c8d9
L-BFGS not implemented in jax
bonneted May 6, 2024
68ab04f
Merge branch 'jax-variableValue' into SPINN-only
bonneted May 6, 2024
32b760f
fix number of sampling points
bonneted May 23, 2024
facfcbf
scale factor for trainable variables
bonneted May 23, 2024
d6f29a9
uniform spinn sampling
bonneted Jun 14, 2024
b74eb4d
undo scale factor
bonneted Jun 17, 2024
9871395
remove save model
bonneted Jun 17, 2024
d9ffe22
Merge branch 'master' into SPINN-only
bonneted Jun 17, 2024
98d414c
update Helmotz example
bonneted Jun 17, 2024
37110d1
snn in a different file
bonneted Jun 17, 2024
719a77b
spinn credit
bonneted Jun 17, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove empty line
  • Loading branch information
bonneted committed May 3, 2024
commit 3d64359158f118d62649d177ed31523283b167eb
1 change: 0 additions & 1 deletion deepxde/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -666,7 +666,6 @@ def _train_sgd(self, iterations, display_every):
for i in range(iterations):
self.callbacks.on_epoch_begin()
self.callbacks.on_batch_begin()

self.train_state.set_data_train(
*self.data.train_next_batch(self.batch_size)
)
Expand Down
Loading