How do you access a trained Modulus Model checkpoint

Once I have trained the model, how do I access the model itself? So, once it has finished training, how would I plug in values for variables trained on into this model?

Hi @isabella.hillman

Your model is saved in a model checkpoint file located in your outputs folder near your training script. With the model checkpoint you load it a python script and it can use it for what you would like. You could do this via the Modulus framework or fall back to a more native PyTorch route.

The Link is not available. Is it still up to date?

Updated link now on Github:

You can also set run_mode: 'eval' in your config YAML.