Figured this out. I traversed the named children of the loaded module and was able to save off the .value of the child module I was interested in. From there I could run a forward pass on the single layer from the model