name 'remove_dataparallel_wrapper' is not defined

[Diyorazh]

Traceback (most recent call last):
File "/Users/Diyora/PnP-SCI_python/pnp_sci_video_grayscale_benchmark.py", line 150, in
state_dict = remove_dataparallel_wrapper(state_dict)
NameError: name 'remove_dataparallel_wrapper' is not defined

I am running the code on python 3.7.6 conda base on the VS Code of my own laptop. I tried to run the code, and I change the use GPU = False. I have already installed a few packages on 'colour' (which I think you may need to state in the dependencies), however, the code still didn't work as expected. It said "'remove_dataparallel_wrapper' is not defined" Thanks!