README.md

metadata

license: mit

import torch

Загрузка весов модели

model_data = torch.load('scorpions.pth')

Просмотр ключей (структуры) словаря модели

print(model_data.keys())