ADD Pix2Pix to SberSwap about ailia-models HOT 5 CLOSED

        x = torch.randn(1, 3, 256, 256, requires_grad=True)
        torch.onnx.export(model.netG,              # model being run
                        x,                         # model input (or a tuple for multiple inputs)
                        "netG.onnx",               # where to save the model (can be a file or file-like object)
                        opset_version=11,          # the ONNX version to export the model to
                        input_names = ['input'],   # the model's input names
                        output_names = ['output'], # the model's output names
                        dynamic_axes={'input' : {0 : 'batch_size'},    # variable length axes
                                        'output' : {0 : 'batch_size'}})