亚马逊AWS官方博客

Tag: DeepJavaLibrary (DJL)

动手用 Java 训练深度学习模型

为了减少 Java 开发者学习深度学习的成本,AWS构建了Deep Java Library (DJL) (https://djl.ai/),一个为 Java 开发者定制的开源深度学习框架。它为 Java 开发者对接主流深度学习框架提供了一个桥梁。DJL 同时对 Apache MXNet,PyTorch 和 TensorFlow 最新版本的支持,使得开发者可以轻松使用Java构建训练和推理任务。在这个文章中,我们会尝试用 DJL 构建一个深度学习模型并用它训练MNIST手写数字识别任务。