mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
added clarifying comments
This commit is contained in:
1 parent
91b3a0aac6
commit
cc12a1c8fa
1 file changed
+5
-1
@@ -609,7 +609,11 @@ class Trainer(TrainerIO):
|
||||
else:
|
||||
output = self.model.training_step(data_batch, batch_nb)
|
||||
|
||||
model_specific_tqdm_metrics_dic = output['tqdm_metrics']
|
||||
try:
|
||||
model_specific_tqdm_metrics_dic = output['tqdm_metrics']
|
||||
except TypeError as e:
|
||||
model_specific_tqdm_metrics_dic = {}
|
||||
|
||||
loss = output['loss']
|
||||
|
||||
self.__add_tqdm_metrics(model_specific_tqdm_metrics_dic)
|
||||
|
||||
Reference in new issue
Block a user