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出来るようになった。
参考
https://www.gitmemory.com/issue/tensorflow/hub/48/460211311
コメント