search

lucastabelini / PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)
https://arxiv.org/abs/2004.10924
MIT License
295 stars 74 forks source link

A consultation on the GPU number #52

Closed Starboy-at-earth closed 3 years ago

Starboy-at-earth commented 3 years ago

Dear Author: Could you please tell me whether the code can run on multiple GPU devices during training, without any further laboring by me?

Starboy-at-earth commented 3 years ago

@lucastabelini

lucastabelini commented 3 years ago

No, not in its current state. However, I believe the only thing you would have to modify is to add something like this line after line 211 of train.py:

model = nn.DataParallel(model)

For more information, please see the documentation.

Starboy-at-earth commented 3 years ago

Okay I will give it a try. Thank you for your in-time reply!!!