How could I get PT file with DCN?

[LegendChen-X]

Please refer to the FAQ in doc and search for the related issues before you ask the question.

Describe the question(问题描述)
When I try to use torch.jit.trace, it always give me the error "ValueError: embedding_dim of SparseFeat and VarlenSparseFeat must be same in this model!", but it works well by just do predict

Additional context
Add any other context about the problem here.

Operating environment(运行环境):

• python version [3.7.3]
• torch version [1.12.1+cu102]
• deepctr-torch version [0.2.9]