这是一个包含Keras Transformer层的Docker镜像。Keras Transformer提供了一组用于构建Transformer模型的预构建层,方便开发者使用Keras构建各种Transformer架构,例如BERT、GPT等。使用此镜像可以方便地进行Transformer模型的训练和推理,无需手动安装和配置相关的依赖项。
docker.io/approachingai/ktransformers:0.2.1