Giters

microsoft / MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to make modifications?

GLivshits opened this issue · comments

commented

Hello. I want to convert mxnet model to pytorch. I've found out bad rename mappings in mxnet_parser.py, which is easily solved. However, it's not obvious how to make new rename functions to nodes with attributes. For example, BatchNorm mx attributes mapped to some different names, and it is not obvious at all how to map arguments for new ops. Where the mapping can be found?