search

brandstetter-johannes / LPSDA

Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
MIT License
47 stars 12 forks source link

large error #3

Closed panglei-math-zju closed 1 year ago

panglei-math-zju commented 1 year ago

Wonderful job! I'm having some problems running the code， KdV, end_time=40, nt=200, nt_effective=200, nx=256, L=128, samples=512, batch_size=64. After 10 epochs, the vlidation error is larger than 1, and the train speed(4070Ti) is slow(maybe it's caused by change num_workers from 4 to 0). Looking forward to your reply.

panglei-math-zju commented 1 year ago

and the utilization rate of gpu is less than ten percent

brandstetter-johannes commented 1 year ago

Hi @panglei-math-zju, thanks a lot! Hmmm I couldn't reproduce this error, using a V100 for training. could you try with num_workers=4?