diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index ffa400a2..50634c2e 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -340,7 +340,7 @@ class Trainer(TrainerIO):
                     # count items in memory
                     # nb_params, nb_tensors = count_mem_items()
 
-                    metrics = self.model.update_tng_log_metrics(self.__tng_tqdm_dic)
+                    metrics = self.model.module.update_tng_log_metrics(self.__tng_tqdm_dic)
 
                     # add gpu memory
                     if self.on_gpu:
@@ -349,7 +349,7 @@ class Trainer(TrainerIO):
 
                     # add norms
                     if self.track_grad_norm > 0:
-                        grad_norm_dic = self.model.grad_norm(self.track_grad_norm)
+                        grad_norm_dic = self.model.module.grad_norm(self.track_grad_norm)
                         metrics.update(grad_norm_dic)
 
                     # log metrics
@@ -404,7 +404,6 @@ class Trainer(TrainerIO):
         model_specific_tqdm_metrics_dic = output['tqdm_metrics']
         loss = output['loss']
 
-        pdb.set_trace()
         self.__add_tqdm_metrics(model_specific_tqdm_metrics_dic)
 
         # backward pass