You have very specific model that returns losses instead of predictions and it also returns them as a dictionary. Usually models return predictions, then loss functions return loss(es) as tensor(s) and you can access it’s value via .item()