pytorch RNN not supported yet

[xuzhiyuan1528]

Convert model with the latest version of ncnn with pytorch==0.4.1 on Mac:

pytorch -> onnx -> ncnn

A simple RNN model,

rnn = nn.RNN(1, 2, 2)
input = torch.randn(5, 3, 1)
h0 = torch.randn(2, 3, 2)
output, hn = rnn(input, h0)
torch.onnx.export(rnn, input, 'rnn.onnx')

use onnx2ncnn tool:

./onnx2ncnn rnn.onnx

Errors:

Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
Shape not supported yet!
Gather not supported yet!
Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
ConstantFill not supported yet!
  # input_as_shape=1
RNN not supported yet!
  # activations 8
  # hidden_size=2
Squeeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
Shape not supported yet!
Gather not supported yet!
Unsqueeze not supported yet!
  # axes 7
Unsqueeze not supported yet!
  # axes 7
ConstantFill not supported yet!
  # input_as_shape=1
RNN not supported yet!
  # activations 8
  # hidden_size=2
Squeeze not supported yet!
  # axes 7

Anybody has tried to convert the pytorch RNN model before?

[xuzhiyuan1528]

not support yet, confirmed by the author.

[SanjunLiu]

pytorch mobilenetv1-ssd: pytorch -> onnx -> ncnn
Unsqueeze not supported yet! [110/1610]

axes 7

Shape not supported yet!
Gather not supported yet!

axis=0

Unsqueeze not supported yet!

axes 7

Unsqueeze not supported yet!

axes 7

Unsqueeze not supported yet!

axes 7

Shape not supported yet!
Gather not supported yet!

axis=0

Unsqueeze not supported yet!

axes 7

Unsqueeze not supported yet!

axes 7

Unsqueeze not supported yet!

axes 7

Shape not supported yet!
Gather not supported yet!

axis=0

Unsqueeze not supported yet!