Sockeye 3.1.18 & 3.1.29 发布,快速、可扩展的深度学习库
Sockeye 是一个基于 Apache MXNet 的快速而可扩展的深度学习库。Sockeye 代码库具有来自 MXNet 的独特优势。例如,通过符号式和命令式 MXNet API,Sockeye 结合了陈述式和命令式编程风格;可以在多块 GPU 上并行训练模型。 目前 Sockeye 发布了 3.1.18 & 3.1.29 版本,分别带来如下改动: [3.1.28] 从 Khandelwal et al., 2021 添加了 kNN-MT 模型, 安装:参见 faiss 文档,推荐通过 conda 安装。 从 sockeye 模型构建 faiss 索引需要两个步骤: 生成解码器状态:sockeye-generate-decoder-states -m [model] --source [src] --target [tgt] --output-dir [output dir] 构建索引:sockeye-knn -i [input_dir] -o [output_dir] -t [faiss_index_signature] ,其中 input_dir 与 sockeye-g...