mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-30 15:00:39 +00:00
fix pp memory issue (#6344)
Co-authored-by: Tong Li <tong.li35271158@gmail.com>
This commit is contained in:
parent
bb6f5d98fc
commit
8992def757
@ -345,7 +345,7 @@ class GRPOConsumer(BaseConsumer):
|
||||
criterion=_criterion,
|
||||
optimizer=self.optimizer,
|
||||
return_loss=True,
|
||||
return_outputs=True,
|
||||
return_outputs=False,
|
||||
)
|
||||
loss = policy_model_outputs["loss"]
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user