I have trained a resnet50 model by freezing all but the final layer (I have also frozen the batchnorm layers of the imagenet backbone during training). Then I save the
state_dict of simply the head and during inference I am loading it in the head part manually.
However, when I do a forward pass on the same image, every time I get a different output; I tried to look everywhere in the forums and in the class lectures but couldn’t figure anything out. Can someone please point out where I might be going wrong?
Thanks & Regards,