From ad69bf98439e9b43d489fc82a0d9ae4e5c7b1dde Mon Sep 17 00:00:00 2001
From: Ritu Yadav <40523539+RituYadav92@users.noreply.github.com>
Date: Sun, 22 Sep 2024 18:40:40 +0200
Subject: [PATCH] Update evaluator.py

---
 engine/evaluator.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/engine/evaluator.py b/engine/evaluator.py
index da653f83..faabcbea 100644
--- a/engine/evaluator.py
+++ b/engine/evaluator.py
@@ -188,6 +188,8 @@ def __init__(self, args, val_loader, exp_dir, device):
 
     @torch.no_grad()
     def evaluate(self, model, model_name='model', model_ckpt_path=None):
+        # TODO: Rework this to allow evaluation only runs
+        # Move common parts to parent class, and get loss function from the registry.
         t = time.time()
         
         if model_ckpt_path is not None: