VisualSearchZeroShot/IVSNtopdown_30_31_array.lua at master · kreimanlab/VisualSearchZeroShot · GitHub this is the code I’m trying to modify. This is the output of the pytorch sub model https://imgur.com/NJh5iC7, and this is the output of the lua code( will post this as a reply in the thread). Funny thing is when i pass the standard 224x224x3 input the pytorch and caffe model work fine, it only outputs different sizes for the smaller sized image of 28x28x3