(ResNet) Support CrossEntropyCriterion Layer in Spark-DL #7
Description
We need to support CrossEntropyCriterion layer to enable resnet. We can reference the torch implementation at https://github.com/torch/nn/blob/master/CrossEntropyCriterion.lua.
{{ message }}