Converted keras model has different parameters

gmalivenko / pytorch2keras

PyTorch to Keras model convertor

https://pytorch2keras.readthedocs.io/en/latest/

MIT License

858 stars 143 forks source link

Converted keras model has different parameters #127

Open mike-zyz opened 3 years ago

mike-zyz commented 3 years ago

The original PyTorch model has no bias (only weights) in certain Convo layers but the converted Keras model does have bias respectively. And the Keras model shows no BatchNorm2D layers compared to the PyTorch model. I am thinking that the code has conducted BN folding. Is there a way to switch off this function so it will be easier to match the weights?

Tailwhip commented 3 years ago

Have the same problem. Any updates?