rtolps comments

Repositories
Issues
Comments

Results 4 comments of


                                            rtolps

How to convert trained CycleGAN model to ONNX, eventually coreML model

Hi! I'm seeking to convert this model to ONNX too. I wouldn't even know where to start though.

export to onnx

I'm wondering this too. I'm trying this in [my own fork](https://github.com/rtolps/Cats2dogs_ONNX) but I'm not very experienced and it's not quite working.

export to onnx

Update: I made a script to convert the model to ONNX: ``` import torch.onnx from torch import nn from utils import * from dataset import ImageFolder from networks import *...

export to onnx

Update I fixed it! just change all torch.var to torch.std (...) ** 2 and it should export an ONNX model!