From a1672d10a3bf40bcb22b69b7b6a0123a894e6c8e Mon Sep 17 00:00:00 2001 From: ashors1 Date: Tue, 22 Apr 2025 15:36:19 -0700 Subject: [PATCH] fix indent Signed-off-by: ashors1 --- nemo_reinforcer/models/policy/dtensor_policy_worker.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/nemo_reinforcer/models/policy/dtensor_policy_worker.py b/nemo_reinforcer/models/policy/dtensor_policy_worker.py index 2c4bd78efd..a2c4b10a07 100644 --- a/nemo_reinforcer/models/policy/dtensor_policy_worker.py +++ b/nemo_reinforcer/models/policy/dtensor_policy_worker.py @@ -351,11 +351,11 @@ def train( dtype=torch.float32, ) - # Update parameters - self.optimizer.step() - self.scheduler.step() + # Update parameters + self.optimizer.step() + self.scheduler.step() - losses.append(torch.tensor(mb_losses).sum().item()) + losses.append(torch.tensor(mb_losses).sum().item()) # Compute global loss across all ranks with torch.no_grad():