KeyError: "BasicVSR: 'BasicVSRPlusPlus is not in the model registry'" #1203
-
Hi, I am training BasicVSR++ from scratch on HPC cluster with following environment info: `> sys.platform: linux
TorchVision: 0.10.0 After stopping the training due to time limit in cluster, I resume the training using config file basicvsr_plusplus_c64n7_8x1_600k_reds4.py : This is while I resumed the training two times ago and it worked without problem. The complete error message: During handling of the above exception, another exception occurred: Traceback (most recent call last): Could you please help? |
Beta Was this translation helpful? Give feedback.
Replies: 4 comments
-
BasicVSR++ should not be in the model registry. It should be in the backbone registry. |
Beta Was this translation helpful? Give feedback.
-
Thanks for your response @ckkelvinchan |
Beta Was this translation helpful? Give feedback.
-
This is a bit strange. How about trying to use the |
Beta Was this translation helpful? Give feedback.
-
Dear @ckkelvinchan I found out that I have removed @BACKBONES.register_module() in basicvsr_pp.py file by mistake, |
Beta Was this translation helpful? Give feedback.
Dear @ckkelvinchan
I found out that I have removed @BACKBONES.register_module() in basicvsr_pp.py file by mistake,
I fix it and it works without problem.
Thank you