Hello @masahi
I have tried using your TraceWrapper and it works on **BertForSequenceClassification** model. Thanks :) --- [Visit Topic](https://discuss.tvm.apache.org/t/how-to-deal-with-prim-dictconstruct/11978/6) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/f476e1fba88aaa7d3a556bcbe18ee72d12e9bd24371469de9f6dc3118fd3d0e9).