search

amazon-science / earth-forecasting-transformer

Official implementation of Earthformer
Apache License 2.0
349 stars 58 forks source link

refactor: use DDPStrategy instead of DDPPlugin #29

Closed gaozhihan closed 1 year ago

gaozhihan commented 1 year ago

Update the implementation of ApexDDPPlugin to inherit from DDPStrategy instead of DDPPlugin to avoid the following deprecation.

ENV_ROOT/lib/python3.9/site-packages/pytorch_lightning/plugins/training_type/ddp.py:20: LightningDeprecationWarning: The `pl.plugins.training_type.ddp.DDPPlugin` is deprecated in v1.6 and will be removed in v1.8. Use `pl.strategies.ddp.DDPStrategy` instead.