attention-mechanism

There are 1909 repositories under attention-mechanism topic.

  • lucidrains/vit-pytorch

    Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

    Language:Python23.9k1602723.4k