search

gmalivenko / pytorch2keras

PyTorch to Keras model convertor
https://pytorch2keras.readthedocs.io/en/latest/
MIT License
858 stars 143 forks source link

AttributeError: Number of inputs is not equal 1 for unsqueeze layer #149

Closed MT010104 closed 1 year ago

MT010104 commented 1 year ago

When I want to transfer the pre-trained models(LeNet5 or VGG16) by myself, an error is always reported.

image

The length of node.input is 2. Can you help me with this?