fix: handle INFONCE_HARD_NEGATIVES as integer if provided #4545
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 type
PR information
在训练 GME Embedding 时,有可能需要在环境变量中提供
INFONCE_HARD_NEGATIVES
。原本的代码遗漏了将其转换为int
类型。In the training of GME Embedding, it may be necessary to provide
INFONCE_HARD_NEGATIVES
via environment variables. The original code missed converting this value to an integer.Experiment results
如果没有这段修改,在 validation 完成后计算 metrics 时将会报错:
If this modification is not included, an error will occur when computing metrics after validation: