Fix parameter initialization so that RELU works
Fix parameter initialization so that RELU works