i have a classification model and need to extract features map from it by freezing some layers and i did it already and got

checkpoints.pth

now i need the features in a pickle file
the file started with
OrderedDict([('pos_embed1', tensor([[[-0.0053, -0.0039,  0.0004,  ...,  0.0012,  0.0061, -0.0006],
         [ 0.0091, -0.00.....