TPUでTensorflow Hubの重みを使おうとしたらUnsuccessful TensorSliceReader constructor: Failed to get matching files on /tmp/tfhub_modules/

ColabのTPUでTensorflow Hubの重みを使おうとしたら表題のエラー

hub.KerasLayer('https://tfhub.dev/google/cropnet/feature_vector/cassava_disease_V1/1')
Unsuccessful TensorSliceReader constructor: Failed to get matching files on /tmp/tfhub_modules/

TPUはGCPにあるファイルしか読み込めない仕様なので、http経由では重みを取得できない。

そのため、一旦ダウンロードしたSavedModelをGCPにアップロードし、バケットのアドレスを指定する必要がある。

hub.KerasLayer('gs://path/to/my/backet/cropnet_feature_vector_cassava_disease_V1_1')

TPUでTensorflow Hubの重みをFine Tuning出来るようになった。

参考

How to use TF Hub on a distributed setting? - hub


コメント

タイトルとURLをコピーしました