• Suraj Patil's avatar
    update unet2d (#1376) · f07a16e0
    Suraj Patil authored
    * boom boom
    
    * remove duplicate arg
    
    * add use_linear_proj arg
    
    * fix copies
    
    * style
    
    * add fast tests
    
    * use_linear_proj -> use_linear_projection
    f07a16e0
attention.py 37.1 KB