search

faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)
Apache License 2.0
336 stars 80 forks source link

Error importing the model #43

Open castelock opened 5 months ago

castelock commented 5 months ago

Good morning,

I'm using your model with fine-tuning but when I run my program in the line from vit_keras import vit it shows the error message ModuleNotFoundError: No module named 'keras.src.engine'. My tensorflow version is 2.16.1 and Keras 3.1.1. Should I install other versions of these libraries or what is it the problem?

Thank you in advance.

hannahwhyatt commented 4 months ago

I had this issue with tensorflow 2.16.1 as well. Using tensorflow 2.8.0 (with python 3.10) solved the issue.