microsoft / unilm

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

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

Open xbc922 opened 1 year ago

xbc922 commented 1 year 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 4 months ago

@xbc922 Hi, Did you make it?