How to Get Runtime and Predcition Value in ldc_2d.py?

Physics Simulation Modulus Physics-ML Model Framework Technical Support (Modulus Only)
1

Using the ldc_2d.py as a base case, how can I do the following:

(i) Get a u value (the output of the NN) on a specific x and y value?
(ii) Get the runtime (prediction time) on a test dataset?

2

Hello @ngeneva, are you able to help me with this? Especially on part (I), are there specific methods I need to call to get the predictions from the specific input values?

1 Like
3

I managed to reconstruct the network using the saved pth file, i.e., flow_network.0.pth in your demo.