Hi,
This is because this callback should be called once all the backprop work has been done.
Such a callback can only be added during the backward pass itself with the current implementation. This is done using the queue_callback method.