在使用别人的模型训练时，出现RuntimeError: The size of tensor a (1024) must match the size of tensor b (3) at non-singleton dimension 3
Open
- Dominant language
- Python
- Stars
- 36.9k
- Forks
- 5.2k
- PR merge metrics
- No merged PRs in 30d
Description
先是使用自己的数据集从零开始训练，然后把别人的模型覆盖本地生成的模型，再运行命令会出现报错
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reproducing the command after replacing the locally generated model with the other model, then compare the two model configurations around loading and training. Done means the reported run completes without the tensor-size mismatch.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100