use OrderedDict to make softmax L2 reg work