neuraloperator / Geo-FNO

Geometry-Aware Fourier Neural Operator (Geo-FNO)
MIT License
174 stars 46 forks source link

Fix the bug in plasticity_3d.py #3

Closed wuhaixu2016 closed 1 year ago

wuhaixu2016 commented 1 year ago

It seems like we need to add the optimizer.step() in line 228 of plasticity_3d.py. Without this line of code, the model cannot be optimized during training.

zongyi-li commented 1 year ago

Thanks for the catch! It was probably mistakenly removed when we cleaned and uploaded the code.

wuhaixu2016 commented 1 year ago

Thanks for your prompt reply.