Inference of DAT model about easyrobust HOT 4 CLOSED

I'm not sure if vit_huge_patch14_224 is fully compatible. As far as we know mae_vit_huge_patch14 uses global pooling and an extra fc_norm. There may have other differences we are not mentioned causing your problem. You can set strict=True to check if all parameters are aligned.

A safe solution of this problem we advice is that doing import easyrobust.models and then setting --model mae_vit_huge_patch14, instead of using vit_huge_patch14_224

from easyrobust.

Insert import easyrobust.models into here, then you can use mae_vit_huge_patch14

from easyrobust.

fixed in 1b2217c

from easyrobust.

Great! Thank you! It solved the problem.

from easyrobust.