Function _GetReverseOrderedGrads

caffe2/python/data_parallel_model.py:1545–1550

Returns the gradients in reverse order (namespace stripped), for the optimal synchronization order.

(model)

def _GetReverseOrderedGrads(model):
    '''
    Returns the gradients in reverse order (namespace stripped),
    for the optimal synchronization order.
    '''
    return list(reversed(model._grad_names))


# A helper function to extract a parameter's name