initialization as was used for the final layer that we removed