From 85e7ee7d1360c5b14a715eacb9f722ee0bd35ed4 Mon Sep 17 00:00:00 2001
From: MaximumEntropy <sandeep.subramanian.1@umontreal.ca>
Date: Wed, 25 May 2022 14:25:20 -0700
Subject: [PATCH] Fix epoch end

Signed-off-by: MaximumEntropy <sandeep.subramanian.1@umontreal.ca>
---
 .../nlp/models/machine_translation/mt_enc_dec_model.py         | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py b/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py
index 4133d596c44b..4ae12f79c59e 100644
--- a/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py
+++ b/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py
@@ -403,6 +403,9 @@ def validation_step(self, batch, batch_idx, dataloader_idx=0):
 
     def eval_epoch_end(self, outputs, mode, global_rank):
         # if user specifies one validation dataloader, then PTL reverts to giving a list of dictionary instead of a list of list of dictionary
+        if not outputs:
+            return
+
         if isinstance(outputs[0], dict):
             outputs = [outputs]