deepxde Is there a function similar to torch's dataparallel in deepxde?

Is there a function similar to torch's dataparallel in deepxde?

Open chenyv118 opened this issue 1 year ago • 1 comments

When the FNN is ([17]+[512]*3+[357]), there are 357 ODEs, and when dde.jacobian is used, the GPU storage (10G) is not enough.

Apr 13 '23 10:04 chenyv118

https://github.com/lululxvi/deepxde/pull/1205 https://github.com/lululxvi/deepxde/pull/1094

Apr 20 '23 17:04 lululxvi