Является ли torch.device torch-scriptable? - PullRequest
0 голосов
/ 21 февраля 2020

TorchScript - это способ создания сериализуемых и оптимизируемых моделей из кода PyTorch. Любую программу TorchScript можно сохранить из процесса Python и загрузить в процесс, в котором нет зависимости Python.

Является ли torch.device torch-scriptable? Можем ли мы использовать torch.script () в модуле, который содержит torch.device?

...