pytorch_lightning 关于 _get_rank的升级公告

报错ImportError: cannot import name ‘_get_rank’ from ‘pytorch_lightning.utilities.rank_zero’

修改下面的代码即可(2023.03)

from lightning_fabric.utilities.rank_zero import _get_rank 

你可能感兴趣的:(pytorch)