updated args

2019-06-26 17:50:09 -04:00 · 2019-06-26 17:50:09 -04:00 · 808e86b17c
parent 71cd8f549d
commit 808e86b17c
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -401,10 +401,10 @@ class Trainer(TrainerIO):
        if self.data_parallel:
            output = reduce_distributed_output(output, len(self.gpus))

-        pdb.set_trace()
        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