keep-loving-pythonのブログ

Pythonを愛し続けたいです(Pythonが流行っている限りですが。。。)

解決策。bert_en_uncased_L-12_H-768_A-12/4, bert_en_uncased_L-12_H-768_A-12/2とかの関連。

解決策。bert_en_uncased_L-12_H-768_A-12/4, bert_en_uncased_L-12_H-768_A-12/2とかの関連。

何をしようとしてのエラーか

環境

windows10
python3.7
GPUなし

サイト

https://www.kaggle.com/code/nayansakhiya/text-classification-using-bert

コード

m_url = 'https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/2'

は、後述のエラーが出るので以下に書き換えると、

m_url = 'https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/4'

エラー

Traceback (most recent call last):
  File "_kore00.py", line 86, in <module>
    model = build_model(bert_layer, max_len=max_len)
  File "_kore00.py", line 63, in build_model
    pooled_output, sequence_output = bert_layer([input_word_ids, input_mask, segment_ids])
  File "C:\Users\XYZZ0\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 926, in __call__
    input_list)
  File "C:\Users\XYZZ0\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\base_layer.py", line 1117, in _functional_construction_call
    outputs = call_fn(cast_inputs, *args, **kwargs)
  File "C:\Users\XYZZ0\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\autograph\impl\api.py", line 258, in wrapper
    raise e.ag_error_metadata.to_exception(e)
ValueError: in user code:

そもそも、

m_url = 'https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/2'

の場合、

OSError: SavedModel file does not exist at: C:\Users\XYZZ0\AppData\Local\Temp\tfhub_modules\ce53fe6769d2ac3a260e92555120c54e1aecbea6/{saved_model.pbtxt|saved_model.pb}

解決策

m_url = 'https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/2' を/4とかに替えるのは、ダメのよう。 では、 ちゃんと、/2を入手する。

https://stackoverflow.com/questions/62674772/how-to-use-tf-hub-models-locally

で紹介されている、以下から取得できましたーー!

"https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/2?tf-hub-format=compressed"

補足

以下で、/2にアクセスすると、/4につながってしまう(ので、、、↑) https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/4