Trouble fine tuning Levit vision models

Surprising! I had tried master version as well but that was not working for me.

I copied the same notebook in a different session and now it’s working. Thank you so much for helping me out! :slightly_smiling_face:

1 Like

Have you managed to fine tune LeVit to some degree of accuracy?

I’ve managed to get Imagewoof accuracy to about 97-98% after fine tuning for 5 epochs without much tweaking.

Interesting. I’ve copied out, used master…still getting the same error with levit_128.