shenweichen / shenweichen/DeepCTR-Torch
How could I get PT file with DCN?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 3.5k
- Forks
- 733
- PR merge metrics
- No merged PRs in 30d
Description
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]
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the linked FAQ and related issues, then reproduce the torch.jit.trace failure using Python 3.7.3, PyTorch 1.12.1+cu102, and deepctr-torch 0.2.9. Trace the DCN model path to determine why the embedding dimensions diverge during tracing; done means a verified resolution or a clearly documented limitation and workaround.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 25/100