both files appear corrupted

I’ve tried to use any of the pkl files you have uploaded, but they all appear corrupted. I tried both as a model and as a weights file:

import torch
import torch.nn as nn

model = torch.load("resnet50_ft_weight.pkl")
print(model)

import torch
import torch.nn as nn
import torchvision.models as models

model = models.resnet50()
model.load_state_dict(torch.load("resnet50_ft_weight.pkl"))
print(model)

The output is:

RuntimeError: Invalid magic number; corrupt file?

[RunOrVeith]

This works:

def resnet50(weights_path=None, **kwargs):
    """Constructs a ResNet-50 model.
    """
    model = ResNet(Bottleneck, [3, 4, 6, 3], **kwargs)
    if weights_path:
        import pickle
        with open(weights_path, 'rb') as f:
            obj = f.read()
        weights = {key: torch.from_numpy(arr) for key, arr in pickle.loads(obj, encoding='latin1').items()}
        model.load_state_dict(weights)
    return model

model = resnet50("/path/to/weights.pkl", num_classes=8631)  # Pretrained weights fc layer has 8631 outputs

[schudov]

try this:

import pickle with open(‘senet50_ft_weight.pkl’, ‘rb’) as f: obj = f.read() weights = pickle.loads(obj, encoding=‘latin1’)