jinns 0.4.1__py3-none-any.whl → 0.4.2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- jinns/solver/_solve.py +2 -3
- {jinns-0.4.1.dist-info → jinns-0.4.2.dist-info}/METADATA +1 -1
- {jinns-0.4.1.dist-info → jinns-0.4.2.dist-info}/RECORD +6 -6
- {jinns-0.4.1.dist-info → jinns-0.4.2.dist-info}/WHEEL +1 -1
- {jinns-0.4.1.dist-info → jinns-0.4.2.dist-info}/LICENSE +0 -0
- {jinns-0.4.1.dist-info → jinns-0.4.2.dist-info}/top_level.txt +0 -0
jinns/solver/_solve.py
CHANGED
|
@@ -196,7 +196,7 @@ def solve(
|
|
|
196
196
|
if carry["param_data"] is not None:
|
|
197
197
|
batch = append_param_batch(batch, carry["param_data"].get_batch())
|
|
198
198
|
carry["params"], carry["opt_state"] = optimizer.update(
|
|
199
|
-
params=carry["params"], state=carry["
|
|
199
|
+
params=carry["params"], state=carry["opt_state"], batch=batch
|
|
200
200
|
)
|
|
201
201
|
|
|
202
202
|
# check if any of the parameters is NaN
|
|
@@ -260,7 +260,6 @@ def solve(
|
|
|
260
260
|
{
|
|
261
261
|
"params": init_params,
|
|
262
262
|
"last_non_nan_params": init_params.copy(),
|
|
263
|
-
"state": opt_state,
|
|
264
263
|
"data": data,
|
|
265
264
|
"curr_seq": curr_seq,
|
|
266
265
|
"seq2seq": seq2seq,
|
|
@@ -281,7 +280,7 @@ def solve(
|
|
|
281
280
|
|
|
282
281
|
params = res["params"]
|
|
283
282
|
last_non_nan_params = res["last_non_nan_params"]
|
|
284
|
-
opt_state = res["
|
|
283
|
+
opt_state = res["opt_state"]
|
|
285
284
|
data = res["data"]
|
|
286
285
|
loss = res["loss"]
|
|
287
286
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: jinns
|
|
3
|
-
Version: 0.4.
|
|
3
|
+
Version: 0.4.2
|
|
4
4
|
Summary: Physics Informed Neural Network with JAX
|
|
5
5
|
Author-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
|
|
6
6
|
Maintainer-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
|
|
@@ -12,11 +12,11 @@ jinns/loss/_operators.py,sha256=HYGDq3K_K7lztDp6al88Q78F6o-hDRo3ODhw8UMRdC8,5690
|
|
|
12
12
|
jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
13
13
|
jinns/solver/_rar.py,sha256=V9y07F6objmP6rPA305dIJ82h7kwP4AWBAktZ68b-38,13894
|
|
14
14
|
jinns/solver/_seq2seq.py,sha256=XNL9e0fBj85Q86XfGDzq9dzxIkPPMwoJF38C8doNYtM,6032
|
|
15
|
-
jinns/solver/_solve.py,sha256=
|
|
15
|
+
jinns/solver/_solve.py,sha256=Yz_asD0ZuYN923E6ysxtPemN-36Zt-BtlSPkTX-BfA8,10047
|
|
16
16
|
jinns/utils/__init__.py,sha256=-jDlwCjyEzWweswKdwLal3OhaUU3FVzK_Ge2S-7KHXs,149
|
|
17
17
|
jinns/utils/_utils.py,sha256=bQm6z_xPKJj9BMCr2tXc44IA8JyGNN-PR5LNRhZ1fD8,20085
|
|
18
|
-
jinns-0.4.
|
|
19
|
-
jinns-0.4.
|
|
20
|
-
jinns-0.4.
|
|
21
|
-
jinns-0.4.
|
|
22
|
-
jinns-0.4.
|
|
18
|
+
jinns-0.4.2.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
|
|
19
|
+
jinns-0.4.2.dist-info/METADATA,sha256=maaTIojnCdHhTIPd2kCJa8QGfX36mWAt1DN7q8Pd_3o,1821
|
|
20
|
+
jinns-0.4.2.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
|
|
21
|
+
jinns-0.4.2.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
|
|
22
|
+
jinns-0.4.2.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|