search

vita-epfl / detection-attributes-fields

PyTorch implementation of "Detecting 32 Pedestrian Attributes for Autonomous Vehicles"
https://arxiv.org/abs/2012.02647
Other
31 stars 6 forks source link

To Run Decoder on GPU #10

Open Karthik-S-EC opened 2 years ago

Karthik-S-EC commented 2 years ago

Hi @svenkreiss @taylormordan.

Is there a way that we can run decoder on GPU? Evaluation and Validation is consuming more time during the decoder run as it runs on CPU.

Thanks in Advance