LookupError: No gradient defined for operation 'pool5' (op type: MaxPoolWithArgmax)

Joker316701882 / Deep-Image-Matting

This is tensorflow implementation for paper "Deep Image Matting"

626 stars 188 forks source link

LookupError: No gradient defined for operation 'pool5' (op type: MaxPoolWithArgmax) #10

Closed Devin147 closed 6 years ago

Devin147 commented 6 years ago

The function tf.nn.max_pool_with_argmax() cannot compute gradient. Did you have the problem? If I want to replace it with tf.nn.max_pool(), what should I do? default