• Jiatao Gu's avatar
    add new_arange function + FIX BUGS of returning attn values · cce92bdd
    Jiatao Gu authored
    Summary:
    Implementation of Levenshtein Transformer paper.
    Add a new helper function "new_arange" to create arange tensor easily.
    Fix bugs of returning attn values for NAT models
    Delete files which are not necessary or experimental.
    
    Reviewed By: kahne
    
    Differential Revision: D17652009
    
    fbshipit-source-id: 436bbb5d45de2f8067003232de4f2bd51e87719c
    cce92bdd
utils.py 13.5 KB