1. 02 Jul, 2020 1 commit
  2. 26 Jun, 2020 1 commit
    • Jintao Lin's avatar
      Add utils to calculate model complexity and adjust folder structure for mmcv/cnn/utils . (#366) · 9bc0b96d
      Jintao Lin authored
      * Add utils to calculate model complexity info
      
      * remove _InstanceNorm in unittest
      
      * add docstring and increase unittest coverage
      
      * fix deconv_flops_counter_hook to accept different data shape
      
      * test when model is not a common instance
      
      * put flops_counter.py and weight_init.py into mmcv/cnn/utils folder
      
      * fix import name
      
      * reformat some docstrings
      
      * update the scripts with latest one and remove redundant codes
      
      * directly represent a model without string and eval()
      
      * reformat code
      9bc0b96d
  3. 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