def get_from_queue(self, trainer: "pl.Trainer", queue: torch.multiprocessing.SimpleQueue) -> None: """Retrieve the :attr:`trainer.callback_metrics` dictionary from the given queue. To preserve consistency, we cast back the data to ``torch.Tensor``. Args: queue: the instance of the queue from where to get the data. """ # NOTE: `add_to_queue` needs to be called before callback_metrics: dict = queue.get() trainer.callback_metrics.update( apply_to_collection(callback_metrics, np.ndarray, lambda x: torch.tensor(x)))
def get_from_queue(self, trainer: Trainer, queue: torch.multiprocessing.SimpleQueue) -> None: self.new_test_val = queue.get() return super().get_from_queue(trainer, queue)
def get_from_queue(self, queue: torch.multiprocessing.SimpleQueue) -> None: self.test_val = queue.get() return super().get_from_queue(queue)