Отслеживание Pytorch - Как вы отслеживаете сохраненный вес?

Question

Я пытаюсь отследить сохраненный вес, чтобы загрузить его с помощью Catalyst. Я проверил документацию по PyTorch, но нашел только трассировку по функциям.

Answer 1

Вы можете отслеживать torch.nn.Module экземпляров.

Если вы не используете управление потоком данных, просто используйте torch.jit.trace, например:

torch.jit.trace(module, example_input)

Если это так, используйте torch.jit.script:

torch.jit.script(module)

Дополнительную информацию можно найти в руководствах или соответствующей документации.