-
Notifications
You must be signed in to change notification settings - Fork 143
Open
Labels
Description
Describe the bug
No batch normalization layers present in the converted keras model.
To Reproduce
Snippet of your code
Expected behavior
A clear and concise description of what you expected to happen.
All layers including the batch normalization layers should be present in the converted keras model
Logs
If applicable, add error message to help explain your problem.
Environment (please complete the following information):
- OS: [e.g. iOS] Ubuntu-20.04
- Python [e.g. Python 2, Python 3] - Python 3.8
- Version [e.g. v0.1.11] - master
Additional context
Add any other context about the problem here.
When converting a pytorch model into keras_model, doing keras_model.summary() does not show the batch normalization layers.