How to Get Runtime and Predcition Value in

Using the 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?

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

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