nDCG can not be called with float targets
nDCG can not be called with float targets: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). 🐛 Bug Invoking RetrievalNormalizedNDCG & retrieval normalized dcg with target of type float results into ValueError: target must be a tensor of booleans or integers The reason is this…
The task
🐛 Bug Invoking RetrievalNormalizedNDCG & retrieval_normalized_dcg with target of type float results into ValueError: target must be a tensor of booleans or integers The reason is this check in _check_retrieval_functional_inputs: To Reproduce The code samples below are…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.