from timm.models import create_model
import utils

model = create_model('checkpoints/repvit_m0_9/2024_05_21_18_42_00/checkpoint_best.pth')
utils.replace_batchnorm(model)