Problem in fitting model with tensorflow.

I am getting error after running following code :

model.fit(x_train, y_train, epochs=10)

After running above code I am getting following error: Epoch 1/10

NotFoundError Traceback (most recent call last) Cell In[11], line 1 ----> 1 model.fit(x_train, y_train, epochs=10) ... ... ... ... ...

Node: 'StatefulPartitionedCall_2' could not find registered platform with id: 0x112e240d0 [[{{node StatefulPartitionedCall_2}}]] [Op:__inference_train_function_763]

Replies

Deppending on your tensorflow-macos and tensorflow-metal you can try import legacy optimizer
from tensorflow.keras.optimizers.legacy import Adam
and use it on model compile stage.