如何在Pytorch中并行运行多个独立输入的网络？

Question

我有一个神经网络，有两个独立的向量作为输入，类似于this question。 两个输入都经过编码，然后进一步处理。但在此之前，这些输入的编码是完全独立的。 如何在pytorch中并行化编码阶段？ 我的代码的一个简单示例：

class MyModel(nn.Module):
    def __init__(params):
        super(MyModel self).__init__()

        self.encoder1 = Encoder1()
        self.encoder2 = Encoder2()

    def forward(x1, x2):
        # how to calculate both encodings in parallel?
        enc1 = self.encoder1(x1)
        enc2 = self.encoder2(x2)
        return some_func(enc1, enc2)

我查看了pytorch论坛，了解了一些DataParallel的知识，但是我不知道如何将其应用到我的案例中，并且只在1个GPU上。你知道吗