Codes to be updated to supports the pytorch-lightning -- 1.9.0

[yiyuzhuang]

Thanks to the author! When I tried it on pytorch-lightning -- 1.9.0， I had met several bugs due to the new features of pytorch-lightning. I would like to share my revised codes here:

• First, the feature "_get_rank" is moved, so you should modify all the code like that:

  # from pytorch_lightning.utilities.rank_zero import _get_rank
  # -->
  from lightning_fabric.utilities.rank_zero import _get_rank  #corrected by yy

• Second, in Line 6 of /utils/callbacks.py , modification should be:

  # from pytorch_lightning.callbacks.base import Callback
  # -->
  from pytorch_lightning.callbacks import Callback

  Note that, it would help to fix the bug named "ValueError("Expected a parent")".

[bennyguo]

Thanks! Fixed in the latest commit! 😁