thanks a lot for replying but will the results with both methods be the same ? and if I want the features for the images from this model but without a classification layer … should I remove or ignore the last two layers (head and avgpool) … as I tried to ignore them but got the features with large size