Преобразование агента PyTorch в TensorFlow (проблема формата) - PullRequest
0 голосов
/ 05 марта 2019

после обучения GAN с использованием PyTorch я получил файл .pth, в котором хранится агент. Я хотел бы загрузить агент в Unity , который, насколько я понимаю, предназначен для приема в качестве агентов ввода из TensorFlow (в формате .nn).

Есть ли способ конвертировать .pth файл в .nn? И / или есть ли способ загрузить агентов, сохраненных как .pth, в Unity? Спасибо!

...