dniku / rl-attention

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement attention model from LTIAA

dniku opened this issue · comments

"Region-sensitive module", figure 2 in https://arxiv.org/pdf/1812.11276.pdf.

Make a copy of policies.MlpPolicy and put it at the point where convolutional layers are translated to dense.