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()