Skip to content
Snippets Groups Projects
Commit e6ff2c55 authored by Zheng Zhang's avatar Zheng Zhang Committed by Kai Chen
Browse files

Code for "an empirical study of spatial attention mechanisms in deep networks" (#730)

* update configs and related code

* remove debug info

* fix style issues

* fix unnecessary changes

* move generalized_attention.py from ops/dcn to models/plugins

* change the arguments name of generalized attention, add related description and update resnet backbone

* revert the change of weight_init.py

* format touched files by yapf

* change default data_root

* update code for passing flake8 test

* delete unnecessary __init__.py

* Change the argument order of resnext.__init__()

* remove constant_init

* simplifying the init of generalized_attention module

* remove unused import

* format the code

* change the name of config folder, add more configs, add README.md

* rename num_head to num_heads

* minor formatting
parent 9df04d54
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment