函数调用堆栈：模型拟合的 train_function 错误

Question

函数调用堆栈：模型拟合的 train_function 错误

model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['accuracy'])
model.fit(X_train_scaled, y_train, epochs=10)

Answer 1

Function call stack: train_function

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

添加此代码