Yu Chen

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

1

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.

AttributeError: type object 'HyperboloidWrappedNormal' has no attribute 'loc'

2

Thanks for your work, but I have a problem when trying your code, the code is currently not working, specifically executing the command: `python main.py --dataset=mnist --batch_size=100 --epochs=100 --model=hyperbolic --wandb...