From 1b497ac69a07ad0caa11a860a5e46125fd41a771 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 20:32:20 -0400 Subject: [PATCH] updated args --- 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 ae7f1392..a485682c 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -395,6 +395,7 @@ class Trainer(TrainerIO): # forward pass # return a scalar value and a dic with tqdm metrics + # TODO: fix training part... loss, model_specific_tqdm_metrics_dic = self.model.training_step(data_batch, batch_nb) self.__add_tqdm_metrics(model_specific_tqdm_metrics_dic)