Fixes for weight and bias logging