RuntimeError: Key _CHECKPOINTABLE_OBJECT_GRAPH not found in checkpoint

[niloofarchzi]

Hi, please help me.
I have been changed your code and add some conv layer in resblock code . i download the model and checkpoints that it was in github file and put them in their correct directory and train this code with no error.
but in the 'test' i got this error :
RuntimeError: Key _CHECKPOINTABLE_OBJECT_GRAPH not found in checkpoint
and i really dont know what to do
please help