How to load viT models in fastai learner?

[m-ali-awan]

TypeError: init() got an unexpected keyword argument 'global_pool'

This error is thrown whenever we use Transformer model:

'''
timm_learner(dls,'vit_small_patch16_224', loss_func=LabelSmoothingCrossEntropy(),metrics=[accuracy,
top_3_accuracy,
f1score,
rocauc,
prec
], opt_func=ranger,model_dir=struggling_models_path)

'''
Thanks for any help

[AndreasNeuhierl]

Same Problem here. Did you find a solution?