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?
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.