search

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
642 stars 259 forks source link

Using multiple GPUs #35

Closed 3rdnightbaron closed 5 years ago

3rdnightbaron commented 5 years ago

@leeyeehoo I was trying to use more than 1 GPU to train.

I did modify the code by adding nn.DataParallel, but it doesnt seem to work.

What is the fix for the issue?