try manual init for softmax