search

AlexanderLutsenko / nobuco

Pytorch to Keras/Tensorflow/TFLite conversion made intuitive
MIT License
272 stars 17 forks source link

Add support for torch.repeat_interleave #17

Closed stakemura closed 11 months ago

stakemura commented 11 months ago

I really appreciate your great work.

Secondly, I found out that current conversion does not support torch.repeat_interleave . It seems that tf.repeat match repeat_interleave.

I would be grateful if repeat_interleave conversion would be supported officially.

AlexanderLutsenko commented 11 months ago

@stakemura Hey, thank you! Added repeat_interleave in v0.10.2

https://github.com/AlexanderLutsenko/nobuco/blob/15673c22cef4065188fc12274a2e34221e6dc7cb/nobuco/node_converters/tensor_manipulation.py#L173

stakemura commented 11 months ago

I confirmed that v.0.10.2 can convert the model including torch.repeat_interleave. I truly appreciate your quick reply!