Ошибка загрузки контрольной точки Pytorch: _pickle.UnpicklingError: неверный ключ загрузки, '\x1f' ⇐ Python

[Anonymous]

Я пытаюсь загрузить веса модели Pytorch, но получаю эту ошибку: _pickle.UnpicklingError: неверный ключ загрузки, '\x1f'.
Вот код загрузки весов:

Код: Выделить всё

import os
import torch
import numpy as np
# from data_loader import VideoDataset
import timm

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print('Device being used:', device)

mname = os.path.join('./CDF2_0.pth')
checkpoints = torch.load(mname, map_location=device)
print("Checkpoint loaded successfully.")
model = timm.create_model('legacy_xception', pretrained=True, num_classes=2).to(device)
model.load_state_dict(checkpoints['state_dict'])
model.eval()

Я пробовал разные версии Pytorch. Я попытался проверить вес, изменив расширение на .zip и открыв его с помощью диспетчера архивов, но не смог решить проблему. Вот общедоступная ссылка на файл весов .pth, который я пытаюсь загрузить. Любая помощь приветствуется!

Подробнее здесь: https://stackoverflow.com/questions/792 ... load-key-x