export model to onnx about photo2cartoon HOT 5 CLOSED

@hao-qiang do you know if the weights can be exported to onnx like here https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html

from photo2cartoon.

was able to get the exported model

from photo2cartoon.

@hao-qiang do you know how to solve the mismatch issue with input output dimensions for the exported onnx model
Information
light : True
dataset : photo2cartoon
batch_size : 1
iteration per epoch : 1000000
Discriminator
discriminator layer : 6
Weight
adv_weight : 1
cycle_weight : 50
faceid_weight : 1
identity_weight : 10
cam_weight : 1000
rho_clipper: 1.0
w_clipper: 1.0
[*] Load models/photo2cartoon_weights.pt Success
torch.Size([1, 3, 256, 256])
torch.Size([1, 2])
torch.Size([1, 1, 64, 64])

see input output size is different

from photo2cartoon.

@hao-qiang any update on this?

from photo2cartoon.

onnx model is uploaded.

from photo2cartoon.