search

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.
MIT License
5.8k stars 965 forks source link

does mmdnn/pytorch support RNN like models? #838

Open Joejiong opened 4 years ago

Joejiong commented 4 years ago

does mmdnn/pytorch support RNN like models?

and i wonder if there is any chat group for mmdnn outside user||developer?

brgds thx

JiahaoYao commented 4 years ago

Hi Joejiong, right now mmdnn hasn't supported the recurrent models. But, probably it will be supported in the near future.

Joejiong commented 4 years ago

Hi Joejiong, right now mmdnn hasn't supported the recurrent models. But, probably it will be supported in the near future.

thx for the answer first.

May I know how do you guys handle problems like RNN op especially pytorch2tf? Do you have any reference doc for me, I also wanna contribute our dygraph to you guys. because there are so many details in doing such commit.haha

thx best regards

JiahaoYao commented 4 years ago

yeah, sounds great! any thoughts from @linmajia @XiaoXYe ?

XiaoXYe commented 4 years ago

Hi @Joejiong ,that sounds great! actually there is a simple RNN process to convert GRU and LSTM in tensorflow parsermmdnn/conversion/tensorflow/tensorflow_parser.py in line 319 : process_graph(self.tf_graph, self.ckpt_data)

GRURewriter and LSTMRewriter we used are inplement in mmdnn/conversion/tensorflow/rewriter/

Joejiong commented 4 years ago

thx