Open noachr opened 1 year ago
Quick changes to get RayStrategy working with latest pytorch and pytorch-lightning. Followed this guide primarily https://lightning.ai/docs/pytorch/stable/upgrade/from_1_6.html
Did not change Horovid and DDPSharded as I am not as familiar with them and they now use separate libaries.
Submitting in case this is useful to anyone.
Quick changes to get RayStrategy working with latest pytorch and pytorch-lightning. Followed this guide primarily https://lightning.ai/docs/pytorch/stable/upgrade/from_1_6.html
Did not change Horovid and DDPSharded as I am not as familiar with them and they now use separate libaries.
Submitting in case this is useful to anyone.