stylegan2-pytorch
stylegan2-pytorch copied to clipboard
FusedLeakyReLUFunction avoid ONNX export
I am trying to export the stylegan generator to onnx but the custom FusedLeakyReLUFunction is not proper for this purpose. Someone can help??? Is this possible?
Same problem, any solution?
me too
Same problem here
ref:https://github.com/yangxy/GPEN/blob/main/face_model/op/fused_act.py#L96