From 91b3a0aac6c038c59f0de17539152af7349a1ade Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 11 Jul 2019 14:57:26 -0400
Subject: [PATCH] added clarifying comments

---
 pytorch_lightning/models/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index cc4fef79..3b376eff 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -628,6 +628,7 @@ class Trainer(TrainerIO):
             for param in model.parameters():
                 print(param.grad.float().sum())
 
+        # avoid memory leaks
         self.batch_loss_value += loss.item()
 
         # gradient update with accumulated gradients