1. 18 Jun, 2020 1 commit
    • Jintao Lin's avatar
      Add `_NonLocalNd` module (#331) · dcc20f3a
      Jintao Lin authored
      * add non_local module
      
      * rewrite non local module comments
      
      * perfect docstring and adjust init function
      
      * not to init norm layer
      
      * Correct initialize when there is a norm
      
      * set normal method for both embedded_gaussian and dot_product
      dcc20f3a