search

gau820827 / AI-writer_Data2Doc

PyTorch Implementation of NBA game summary generator.
Apache License 2.0
83 stars 22 forks source link

[gau820827] Fix effciency in Attn module and reordering #34

Closed gau820827 closed 6 years ago

gau820827 commented 6 years ago
  1. Rewrite the for loop inside Attn module.
  2. Reorder the input information to a more reasonable aggregation.