merantix / imitation-learning

Autonomous driving: Tensorflow implementation of the paper "End-to-end Driving via Conditional Imitation Learning"
https://medium.com/merantix/journey-from-academic-paper-to-industry-usage-cf57fe598f31
MIT License
89 stars 21 forks source link

Speedup training through num_parallel_calls #10

Closed markus-hinsche closed 5 years ago

markus-hinsche commented 5 years ago

Make training faster by using num_parallel_calls parameter: This means that the input pipeline's preprocessing will run computation in parallel now.

The speedup depends on the hardware obviously, but we measured a 2.6x speedup for our setup.