Error while printing the summary of a pre-trained model (Transfer Learning Section)

[E-Khalouei]

Hello all,

I have completed the Transfer Learning section and decided to apply the code from this section to a pre-trained model. The input for this model is an audio dataset. However, I encountered an issue when loading the model using torch.load. The parameters and weights do not seem to load explicitly, and when I try to print the model's summary, an error occurs.

I would like to ask for guidance on how to properly load a pre-trained model from Zenodo that is trained on an audio dataset.

Here is the code I am using:

import torch
model = torch.load('Cnn10_mAP=0.380.pth', map_location=torch.device('cpu'))

he model can be found at: https://zenodo.org/records/3987831