#2549·OpenNMT-py

在 onmt.utils.distributed.all_reduce_and_rescale_tensors 中,列下标超出范围:51

作者: alexis-allemann创建于 2024年1月11日更新于 2024年1月11日

当使用 `parallel_mode="data_parallel"` 在多个 GPU 上训练模型时,如果出现"CUDA 内存不足"错误,则会在 `onmt.utils.distributed.all_reduce_and_rescale_tensors` 函数中触发异常:

内容来源: OpenNMT/OpenNMT-py