Error: Cannot import name '_init_vit_weights' from 'timm.models.vision_transformer'

[lxy-146]

Hello, we encountered an error ‘Cannot import name '_init_vit_weights' from 'timm.models.vision_transformer’ while trying to replicate your method. This might be due to changes in the timm version that have altered or removed _init_vit_weights. Could you please provide the version of timm that you used?

[lxy-146]

I found that timm=0.5.4 is functional and hope this helps others.