Change from resnet152 to resnet50
I have perform an image duplication detection withpytorch pretrained model. I am usingresnet152for extracing embedding vector. I already embedding millions of record already. Andresnet152is too big which consume 16GB of GPU memory. I want to switch toresnet50which use only 4GB of GPU memeory. The problem for me is they are not generate the same embedding output. Any suggestion that I can approach this better?