-
Notifications
You must be signed in to change notification settings - Fork 45.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
No OpKernel was registered to support Op 'TPUOrdinalSelector' used by node TPUOrdinalSelector #7501
Comments
What's the TF version you use? This feature is available in 1.14 and later versions. |
I am using 1.14.0 I've been doing transfer learning with ssd_resnet_50_fpn_coco as the base, using the object detection api model_main.py script. Edit: I tried this on my local computer and on a google collab notebook with TPU runtime enabled. No success with either environment. |
I'm getting the same error, with TF 1.14 and 1.15 |
Same issue. TF 1.14, 1.15, tf-nightly |
I was never able to get the
then compiling the model with: https://coral.ai/docs/edgetpu/compiler/ |
@bourdakos1 Whats model you were exporting? |
@nkise ssd mobilenet v1 |
@bourdakos1 Thanks! What Tf version have you used? |
I haven’t thoroughly tested both, but 1.14 and 1.15 should both work |
System information
I have trained a model on my GPU and want to export the model to a TPU for inference.
I am running the script found at
When I run the below command:
python3 object_detection/tpu_exporters/export_saved_model_tpu.py --pipeline_config_file=object_detection/output/models/model/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config --ckpt_path=object_detection/output/models/model/model.ckpt-39688 --export_dir=/tmp/out --input_type="image_tensor" --input_placeholder_name=image_tensor:0
I get the following exception:
It looks like @shreyaaggarwal encountered the same issue, posting it here #4283
The text was updated successfully, but these errors were encountered: