Getting "Runtime node closed unexpectedly - no connection" on example tpu notebook

[cocoa-xu]

Getting "Runtime node closed unexpectedly - no connection" on example tpu notebook

[dkuku]

Thanks @cocoa-xu
It works 🥳, Thanks.

[cocoa-xu]

Hi @dkuku, which version of tflite_elixir was it in the Notebook dependencies and setup cell? And could you please let me know if you're on an x86_64 or aarch64 Linux? Thanks!

[dkuku]

I was using the latest master on intel. In the deps there is 0.1.5. On cpu it runs fine for me

[cocoa-xu]

Oh I kinda know what went wrong here. I haven't added reference counting mechanism to the EdgeTPUContext and related things. So Erlang dropped some resources while they are meant to live at least as long as the Interpreter.

I'll fix this in the next version.

With that said, if you would like to test and see if it can communicate with your TPU, then the normal model also works on TPU (but not optimised). For example,

$ mix classify_image --model test/test_data/mobilenet_v2_1.0_224_inat_bird_quant.tflite \
  --input test/test_data/parrot.jpeg \
  --labels test/test_data/inat_bird_labels.txt \
  --use-tpu \
  --tpu usb
----INFERENCE TIME----
78.4ms
-------RESULTS--------
Ara macao (Scarlet Macaw): 0.70703

[dkuku]

this works 👍🏻

mix classify_image --model test/test_data/mobilenet_v2_1.0_224_inat_bird_quant.tflite \                                                                             1 ✘  1 ☑  3  
  --input test/test_data/parrot.jpeg \
  --labels test/test_data/inat_bird_labels.txt \
  --use-tpu \
  --tpu usb
----INFERENCE TIME----
713.8ms
-------RESULTS--------
Ara macao (Scarlet Macaw): 0.70703

[cocoa-xu]

Hi @dkuku, v0.1.6 is out and it should work as expected!