Cleanup

NVIDIA · Apr 15, 2024 · 3fb59ec · 3fb59ec
1 parent c3c397c
commit 3fb59ec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nemo/core/optim/distributed_adam.py b/nemo/core/optim/distributed_adam.py
@@ -481,6 +481,7 @@ def _param_copy_fragments(self, fragments: Iterable[DistributedFusedAdam.Paramet
         _multi_tensor_copy(
             buffers_in, buffers_out, dummy_overflow_buf=self._dummy_overflow_buf,
         )
+
         # Update transpose caches
         params = set(self.parameter(fragment) for fragment in fragments)
         for param in params: