INNER CODE UNIT · Python

log_probs

nv-tlabs/ATISS · scene_synthesis/losses/__init__.py:112

    log_probs = log_probs + F.log_softmax(logit_probs, -1)
    return -log_sum_exp(log_probs)

View source record →