Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Sid Sahai's avatar
    [Tests] Test attention.py (#368) · f73ca908
    Sid Sahai authored 
    * add test for AttentionBlock, SpatialTransformer

* add context_dim, handle device

* removed dropout test

* fixes, add dropout test
    f73ca908
test_layers_utils.py 12.3 KB