def predict(self):
"""
AUTHORS:
--------
:author: Samuel Westlake
:author: Alix Leroy
DESCRIPTION:
------------
Inference of the model
PARAMETERS:
-----------
None
RETURN:
-------
:return outputs->dict: the total losses and total metrics for the model over the test data set
"""
# Initialise an empty tensor to store outputs
outputs = Tensor()
for minibatch_index, minibatch in enumerate(self.dataloader, 0):
inputs, labels, additional_data = self.clean_single_element_list(minibatch)
# Infer the outputs from the model over the given mini batch
minibatch_output = self.model(*inputs)
# Append mini_batch output to the output tensor
outputs.cat(minibatch_output.detach())
return outputs
