From 6a6ca3b299e5dd1afa0cfd9366ceb10ba4e180ae Mon Sep 17 00:00:00 2001 From: Edgar Riba Date: Thu, 17 Jun 2021 15:37:43 +0200 Subject: [PATCH] Update pytorch_lightning/core/lightning.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Carlos Mocholí --- pytorch_lightning/core/lightning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 72d3debc369c5..c5c83880d6249 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1971,7 +1971,7 @@ def get_from_queue(self, queue: torch.multiprocessing.SimpleQueue) -> None: Args: queue: the instance of the queue from where to get the data. """ - # NOTE: this must be called in the right order to get the `callback_metrics` + # NOTE: `add_to_queue` needs to be called before callback_metrics: dict = queue.get() self.trainer.callback_metrics.update( apply_to_collection(callback_metrics, np.ndarray, lambda x: torch.tensor(x))