can developer add change basemodel class device attribute?

[myhaa]

I trained the model on the GPU and predicted the model on the CPU. An error was reported。

[zanshuxun]

Please add more infomation for us to reproduce the error.

Describe the bug(问题描述) A clear and concise description of what the bug is.

To Reproduce(复现步骤) Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Operating environment(运行环境):

• python version [e.g. 3.6, 3.7]
• torch version [e.g. 1.9.0, 1.10.0]
• deepctr-torch version [e.g. 0.2.9,]

Additional context Add any other context about the problem here.