I need a retrieval system which is based on a pretrained model for the CIFAR dataset.

Use the last fully connected layer features from this model and implement a nearest neighbour based image retrieval system. Also report retrieval metrics like AP. You can use publically available pretrained weights, as well as other code as reference. Should be done in Pytorch. 
Please comment your code so it is easily understandable.