与nngraph Torch并行计算

时间:2016-05-27 20:08:22

标签: neural-network torch

我认为nngraph应该能够更复杂地编写具有复杂结构(例如,并行计算)的神经网络,但我遇到了一些错误..

nngraph的主要优点是能够在之后绘制图形,还能够轻松地链接模块吗?

为什么我有这个错误:

lookup = nn.LookUpTable(...)
question = lookup(input[1])
answer = lookup(input[2])

或者我应该做点什么

question, answer = lookup({input[1],input[2]}

(输入[1]和输入[2]只是包含整数的张量) (错误是问题或答案之一没有正确的输出。尺寸关闭等)。

即使在使用nngraph时我是否必须使用ParallelTable?在这种情况下?

1 个答案:

答案 0 :(得分:1)

好吧,显然这与nngraph无关。

在这种情况下,您应该做的是克隆您的模块,并使用克隆:

lookup2 = lookup:clone()

lookup和lookup2将共享相同的参数。