assert tensor_dict1.batch_size == tensor_dict2.batch_size error

执行 sh scripts/rl.sh
File "*****/cudaLLM_rl/main_ppo.py", line 561, in run
trainer.fit()
File "*****/cudaLLM_rl/ray_trainer.py", line 892, in fit
batch = batch.union(gen_batch_output)
File "*****/verl/verl/protocol.py", line 620, in union
self.batch = union_tensor_dict(self.batch, other.batch)
File "*****/verl/verl/protocol.py", line 107, in union_tensor_dict
assert tensor_dict1.batch_size == tensor_dict2.batch_size, (
AssertionError: Two tensor dict must have identical batch size. Got torch.Size([2048]) and torch.Size([256])

唯一的改动就是把n_nodes改成了1，这是什么原因