2023年2月6日
使用JAX实现完整的Vision Transformer
摘要： 本文将展示如何使用JAX/Flax实现Vision Transformer (ViT)，以及如何使用JAX/Flax训练ViT。 Vision Transformer 在实现Vision Transformer时，首先要记住这张图。 以下是论文描述的ViT执行过程。 从输入图像中提取补丁图像，并将其 阅读全文
posted @ 2023-02-06 12:42 deephub 阅读(37) 评论(0) 推荐(0) 编辑