Use SelfAttention in relformer_imagenet64.gin (#28)
Use SelfAttention in relformer_imagenet64.gin (#28)

* Use SelfAttention in relformer_imagenet64.gin

* Use SelfAttention in relformer_imagenet64.gin
google/trax,google/trax