Hi, I am getting nan values throughout when I use the pretrained model. Is there anything that needs to be changed? 