Как я могу увидеть прямую функцию модели, которая устанавливается путем загрузки файла .pth?

Question

У меня есть файл .pth, и я могу просто построить сеть, используя model = torch.load("path to .pth") Но я не знаю, как это происходит. Есть ли способ проверить прямую функцию модели? Thx.

Answer 1

Я предполагаю, что ваш .pth файл содержит всю дампированную модель, а не только данные о состоянии, поэтому вы столкнулись с этой дилеммой. Но в любом случае, да, вы можете проверить функцию пересылки, используя библиотеку inspect.

import inspect
import torch

unknown_model = torch.load('save_path.pth')
print(inspect.getsource(unknown_model.forward))