[trainer] fix bug when batch size=1 #3960
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
LemonNoel
commented
Nov 30, 2022
|
|
||
| logits = nested_detach(logits) | ||
| if len(logits) == 1: | ||
| if isinstance(logits, (list, tuple)) and len(logits) == 1: |
Contributor
Author
There was a problem hiding this comment.
这里增加了类型检查，其他修改均为代码风格检查。
sijunhe
pushed a commit
to sijunhe/PaddleNLP
that referenced
this pull request
Nov 30, 2022
sijunhe
added a commit
that referenced
this pull request
Dec 14, 2022
* [NewFeatures]add ci & pypi workflow (#3578) * add basic ci & pypi workflow * add Makefile file to enable more custom scripts * update makefile & workflow * add manifest.in file * save requirements * improve makefile * add test running command * complete simple CI test script * complete simple CI test script * complete first version of workflow * failure at first * remove CI workflow * update workflows * add dev-dependency * complete format & lint * add paddle dependency * complete format & lint & test command * upgrae workflow files * use pytest to do test * update need keywords * change the description * update lint script * upgrade workflow * update parameterized * update makefile * fix lint * update test command in workflow * fix __init__ lint * ignore all __init__.py file * fix script workflow * udpate flake8 config * update flake config * update flake8 config * remove redefinition of unused Co-authored-by: Sijun He <[email protected]> * Add multi recall of semantic search for pipelines (#3864) * Add multi recall of semantic search for pipelines * Update multi recall semantic search README.md * remove unused imports * remove unused imports * Update __init__.py * remove unused imports * restore __init__.py * skip retriever __init__.py * [trainer] fix bug when batch size=1 (#3960) * [PPdiffusers] Release ppdiffusers 0.6.3 (#3963) * release 0.6.3 * release 0.6.3 * style * code style * fix test * remove commented code Co-authored-by: 骑马小猫 <[email protected]> Co-authored-by: w5688414 <[email protected]> Co-authored-by: Noel <[email protected]> Co-authored-by: yujun <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Bug fixes
PR changes
APIs
Description
logits[0]when batch_size=1, which might lead toOut of memory error.