When I run your code (a little modified for config) it gives the following error:
InvalidArgumentError: Cannot assign a device for operation ROI/rpn_non_max_suppression/NonMaxSuppressionV3: Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available. Registered kernels: device='CPU'; T in [DT_FLOAT] device='CPU'; T in [DT_HALF]
When I run your code (a little modified for config) it gives the following error:
Maybe the problem arise due these lines:
How can I solved?