How to convert REPVGG models for inference time usage ?

[Enescigdem]

Hello,
I trained a repvgg model and tried to convert it to an inference time model via convert.py in https://github.com/DingXiaoH/RepVGG repo .
While loading the checkpoint, a lot of unexpected key errors arose.
What could be wrong?
Thanks in advance

[yfor1008]

@Enescigdem , Did you solve the problem? I have the problem too.

[rwightman]

@Enescigdem @yfor1008 an implementation of switch_to_deploy is welcome for the repvgg here, when it was first ported the inference conversion was more messy and intrusive so I skipped it... even better an FX transform version would be welcome but that is more complex for those unfamiliar with FX