nn.DataParallelTableは、カスタム層

Question

マルチGPUモデルの初期化コードで失敗した

FATAL THREAD PANIC: (read) /home/daniel/torch/install/share/lua/5.2/torch/File.lua:343: unknown Torch class <nn.Reorg> 
FATAL THREAD PANIC: (read) /home/daniel/torch/install/share/lua/5.2/torch/File.lua:343: unknown Torch class <nn.Reorg> 

<nn.Reorg>ちょうどそのmodels/Reorg.luaで定義されている私のカスタム層であり、レイヤー入力に対して単純なコピー操作を行います。
CPUおよびシングルGPUの場合は正常に動作します。

Answer 1

私は最終的にちょうどコードの1行を追加し、それを把握：

:threads(function() 
      require 'models/Reorg' 
      local cudnn = require 'cudnn' 
      cudnn.fastest, cudnn.benchmark = fastest, benchmark 
     end) 

のスレッドが、私はそれを手動でロードする必要が（なぜ???）ファイルをロードしませんでしたが...