This is Abishek's implementation of spatial softmax for tf models. I'll experiment with it tonight to see if it works. @cbfinn since you wrote the original spatial softmax and I've not seen it, can you take a look at this and make sure it seems okay?
This is Abishek's implementation of spatial softmax for tf models. I'll experiment with it tonight to see if it works. @cbfinn since you wrote the original spatial softmax and I've not seen it, can you take a look at this and make sure it seems okay?