Mathanraj_Sharma:

model = models.resnet50(pretrained=False)


Yes, resnet50 will yield a flattened activation output of [batch_size, 2048].
However, this is independent to freezing the parameters or training them.