image classification models pytorch. As we have already written the helpe