The code for batch_normalization in architecture.py is wrong
The code for batch_normalization in architecture.py is wrong