Fixed a bug where the softmax layer's parameters were not being set