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.