microsoft / unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
https://aka.ms/GeneralAI
MIT License
20.2k stars 2.55k forks source link

dit cascade model how to export to torchscript or onnx #911

Open xbc922 opened 2 years ago

xbc922 commented 2 years ago

Describe Model I am using (UniLM, MiniLM, LayoutLM ...): dit

i try export dit object detection cascade model to torchscript or onnx but i fail,

i got some error:

KeyError: 'Non-existent config key: MODEL.VIT

yellowjs0304 commented 9 months ago

@xbc922 Hi, Did you make it?