zhanghang1989 / zhanghang1989/PyTorch-Encoding
Questions on the middle output
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 2k
- Forks
- 448
- PR merge metrics
- No merged PRs in 30d
Description
Hi, I am using it to train my own model.
For instance, my model has two out puts (return xup3,x_out).
But when i check it the out put number of my model is always the number of
GPU I used. Since I need to use this output as another input to another model.
Can you please help me with this?
Thank you!
Contributor guide
No contributing guide indexed for this repository
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
No source files or tests are named. Reproduce the reported case with a model returning xup3 and x_out, then trace how the model output count is determined relative to the number of GPUs; done means the outputs match the model's declared returns rather than the GPU count.
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
- Needs clarification
- Newbie friendliness
- 25/100