├── LICENSE ├── README.md ├── blocks.py ├── fig ├── attn_map_after.png ├── attn_map_before.png ├── learn_to_pay_attn.png ├── loss_attn_after.png ├── loss_attn_before.png ├── loss_compare.png ├── test_acc_attn_after.png ├── test_acc_attn_before.png └── test_acc_compare.png ├── initialize.py ├── model1.py ├── model2.py ├── train.py └── utilities.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/README.md -------------------------------------------------------------------------------- /blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/blocks.py -------------------------------------------------------------------------------- /fig/attn_map_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/attn_map_after.png -------------------------------------------------------------------------------- /fig/attn_map_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/attn_map_before.png -------------------------------------------------------------------------------- /fig/learn_to_pay_attn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/learn_to_pay_attn.png -------------------------------------------------------------------------------- /fig/loss_attn_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/loss_attn_after.png -------------------------------------------------------------------------------- /fig/loss_attn_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/loss_attn_before.png -------------------------------------------------------------------------------- /fig/loss_compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/loss_compare.png -------------------------------------------------------------------------------- /fig/test_acc_attn_after.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/test_acc_attn_after.png -------------------------------------------------------------------------------- /fig/test_acc_attn_before.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/test_acc_attn_before.png -------------------------------------------------------------------------------- /fig/test_acc_compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/fig/test_acc_compare.png -------------------------------------------------------------------------------- /initialize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/initialize.py -------------------------------------------------------------------------------- /model1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/model1.py -------------------------------------------------------------------------------- /model2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/model2.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/train.py -------------------------------------------------------------------------------- /utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SaoYan/LearnToPayAttention/HEAD/utilities.py --------------------------------------------------------------------------------