We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7803632 commit ee050e4Copy full SHA for ee050e4
1 file changed
mova/engine/trainer/accelerate/accelerate_trainer.py
@@ -276,7 +276,7 @@ def _setup(self):
276
if self.is_main_process:
277
os.makedirs(self.save_path, exist_ok=True)
278
279
- torch.cuda.memory._record_memory_history(enabled='all')
+ # torch.cuda.memory._record_memory_history(enabled='all')
280
281
def _setup_lora(self):
282
"""Configure LoRA"""
0 commit comments