Inference Problem

[Heba-Atef99]

I am trying to train the model on new data, after training I save the model into .pb file using save.py and create_pb.py but I am facing a problem, whenever I use the new .pb file I get this error

InvalidArgumentError: Graph execution error:

Op type not registered 'nms_map_while_cond_2048' in binary running on 3cece289dc39. Make sure the Op and Kernel are registered in the binary running in this process. Note that if you are loading a saved graph which used ops from tf.contrib, accessing (e.g.) tf.contrib.resampler should be done before importing the graph, as contrib ops are lazily registered when the module is first accessed. while building NodeDef 'import/nms/map/while/cond/_0'

Note that I am not using tf.contrib as it's deprecated so I use tf.slim instead Can you please help me with this error? I don't understand the cause of it or how to solve it