drivendataorg / zamba

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos
https://zamba.drivendata.org/docs/stable/
MIT License
118 stars 27 forks source link

Support `auto_lr_find` with `num_workers` > 0 #227

Open ejm714 opened 2 years ago

ejm714 commented 2 years ago

auto_lr_find does not support DDP. We set the multiprocessing context based on the number of workers. This leads to a scenario where a user cannot use auto_lr_find with multiple workers, even though this would be possible if the multiprocessing context were set to None.

A couple options: