Проблема преобразования PyTorch в ONNX - torch.randn - PullRequest
Новая
       11

Проблема преобразования PyTorch в ONNX - torch.randn

0 голосов
/ 09 января 2020

Используя это переопределение StyleGAN в PyTorch, я пытаюсь экспортировать генератор как файл .onnx, используя следующий код: 

import model
import torch

Gen = model.StyledGenerator(512)
dummy_input = torch.randn(1,512)
model1 = Gen

torch.onnx.export(model1, dummy_input, "disc.onnx", verbose=True)

Когда я запускаю скрипт, Я получил ошибку типа (см. Ниже) из-за перевода torch.randn. У вас есть предложения, как это исправить? 

TypeError: i_(): incompatible function arguments. The following argument types are supported:
    1. (self: torch._C.Node, arg0: str, arg1: int) -> torch._C.Node

Invoked with: %227 : Tensor = onnx::RandomNormal(), scope: StyledGenerator
, 'shape', 222 defined in (%222 : int[] = prim::ListConstruct(%218, %219, %220, %221), scope: StyledGenerator
) (occurred when translating randn)

Чтобы дублировать проблему, вам необходимо скачать / клонировать репозиторий github и компакт-диск в эту папку.

...