TensorFlow/Transformers兼容性问题:导入tensorflow.keras引发TFElectraModel运行失败
TensorFlow/Transformers兼容性问题:导入tensorflow.keras引发TFElectraModel运行失败
最近我遇到了个棘手的问题——两年前写的代码在当前环境里跑会报错,我怀疑这和Keras 3的版本更新有关,想弄明白到底出了什么状况。
目前我找到了两种能让代码正常运行的办法:
- 去掉所有
from tensorflow.keras import something这类导入语句,全程改用tf.keras.something的调用方式 - 直接切换回之前的旧环境运行
这个代码在以下旧环境配置下是完全正常的:
- TensorFlow版本:2.13.0
- Keras版本:2.13.1
- Transformers版本:4.33.3
- Python版本:3.8.10
但在当前环境下,如果我使用from tensorflow.keras import something这种导入方式,代码就会报错,当前环境的版本配置是:
- TensorFlow版本:2.16.1
- Keras版本:3.3.2
- Transformers版本:4.39.3
- Python版本:3.11.5
import tensorflow as tf from transformers import ElectraTokenizer, TFElectraModel from tensorflow.keras.optimizers import Adam # 只要包含这行(不管后续内容)就会报错
备注:内容来源于stack exchange,提问作者qwertzuiop
相关产品推荐
相关产品推荐

