Pytorch Multiprocessing зависает - PullRequest
       8

Pytorch Multiprocessing зависает

0 голосов
/ 28 февраля 2019
for conv_layer in self.conv_layers:
    test_mat = conv_layer(test_mat)
    self.conv_norm_layers.append(nn.BatchNorm2d(test_mat.shape[1]))

Моя многопроцессорная обработка pytorch зависает во время вышеприведенного кода.Не многопроцессорная работа работает без нареканий.Ниже приведен код для запуска многопроцессорной обработки:

for idx in range(num_worker):
    work = mp.Process(
        target=run_experiment,
        args=(experiment,),
        kwargs=dict(...
        )
    )
    work.start()
    works.append(work)

for idx in range(num_worker):
    works[idx].join()
...