Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Patrick von Platen's avatar
    [Reformer] - Cache hidden states and buckets to speed up inference (#5578) · 9d37c56b
    Patrick von Platen authored 
    * fix merge rebase

* add intermediate reformer code

* save intermediate caching results

* save intermediate

* save intermediate results

* save intermediate

* upload next step

* fix generate tests

* make tests work

* add named tuple output

* Apply suggestions from code review

* fix use_cache for False case

* fix tensor to gpu

* fix tensor to gpu

* refactor

* refactor and make style
    9d37c56b
test_modeling_reformer.py 47 KB