fix: improve batch size handling in GRPOTrainer compute_loss#4

All4Nothing wants to merge 1 commit intoFusionBrainLab:mainFusionBrainLab/Vision_GRPO:mainfrom

All4Nothing:fix/batch-size-handlingAll4Nothing/Vision_GRPO:fix/batch-size-handlingCopy head branch name to clipboard