Baseline submission failing though local evaluation working

[kishan_shukla]

My Baseline solution is failing on submission, and I cannot find the issue in the logs. Local Evaluation is working fine. The only change I made to the code is the following (that too because the code was throwing an error without the change):

Here are the logs:
logs.zip (31.7 KB)

Random Ranker submission worked fine. Please help us, as it’s blocking us from moving ahead.

Thanks a lot!

[saidinesh_pola]

Change gpu = true in aicrowd.json if the submission needs gpu