search

faustomorales / vit-keras

Keras implementation of ViT (Vision Transformer)
Apache License 2.0
329 stars 78 forks source link

error with visualization #32

Closed sutummala closed 2 years ago

sutummala commented 2 years ago

There seems to be an error in the visualization file at grid_size = int(np.sqrt(model.layers[5].output_shape[0][-2] - 1)). It is hard coded and seems not suitable for 224 x 224 resolution? Any help is appreciated.

faustomorales commented 2 years ago

Hi there -- This sounds like a problem with the code but it is impossible to trace the issue with the information you have provided. For that reason (lack of information), I am closing this issue. If you provide a minimal code snippet that reproduces the problem, I am happy to re-open the issue. Please note that, the more succinct and targeted your code snippet, the more likely it is that I or another user will be able to help resolve the problem.