ddp fail with new pl_lightning / pytorch

Visual-Behavior / aloception-oss

Aloception is a set of package for computer vision: aloscene, alodataset, alonet.

Other

92 stars 7 forks source link

ddp fail with new pl_lightning / pytorch #274

Open ragier opened 2 years ago

ragier commented 2 years ago

@thibo73800 To reproduce, you can launch a train (detr for example) with this modification if you have only one GPU :

Replace : accelerator=None if torch.cuda.device_count() <= 1 else "ddp", by accelerator="ddp",

anhtu293 commented 1 year ago

It's fixed in the latest version of aloception-oss (v0.5.1). Since v1.8 of lightning, it doesnt accept ddp as value for accelerator anymore. We have strategy for that instead.