标签： python machine-learning parallel-processing neural-network pytorch
我有一个神经网络，其中有两个单独的向量作为输入，类似于this question。 两个输入都经过编码，然后进一步处理。但是直到那时，这些输入的编码是完全独立的。 如何在pytorch中并行化编码阶段？ 我的代码的一个最小示例：
a.yaxis.offsetText.set_visible(False)
我检查了pytorch论坛并了解了一些有关DataParallel的知识，但是我不确定如何仅在1个GPU上使其适合我的情况。