Commit 6e20f3bd authored by Piotr Czapla's avatar Piotr Czapla Committed by Rasmus Munk Larsen
Browse files

Added training parameter to batch_normalization (#16134)

* Added training parameter to batch_normalization

According to the docs the batch_normalization layer does not work properly if the parameter is not set correctly.

* Clean up multiple tests for training mode
parent 48178e04
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment