bug fixes

1 job for weight_initializer in 1 minute and 21 seconds