Make lightning reproducible

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

https://detectron2.readthedocs.io/en/latest/

Apache License 2.0

29.3k stars 7.32k forks source link

Closed ayushidalmia closed 2 months ago

ayushidalmia commented 2 months ago

Summary: In this diff we make changes to ensure we can control reproducibility in d2go:

 {
                "sync_batchnorm": True,
                "deterministic": True,
                "replace_sampler_ddp": False,
 }

allow passing prefetch_factor, pin_memory, persistent_memory as args to batch dataloader.
minor fix in training sampler

Differential Revision: D55767128

facebook-github-bot commented 2 months ago

This pull request was exported from Phabricator. Differential Revision: D55767128

facebook-github-bot commented 2 months ago

This pull request has been merged in facebookresearch/detectron2@bce6d7262b1065498481be1d6708c8dbb142975a.