/SimpleSelfAttention

A simpler version of the self-attention layer from SAGAN, and some image classification results.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Watchers