Commit ca1aa4ad authored by Nand Dalal's avatar Nand Dalal Committed by Martin Wicke
Browse files

Allow using custom initializers for batch norm params (#3960) 

* Allow using custom initializers for batch norm params
* Replace with initializers dict for beta, gamma, moving_mean, and moving_variance
* Added custom initializers test case for slim batch norm
parent 2451221e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment