Update evaluator.py

VMarsocci · Sep 22, 2024 · ad69bf9 · ad69bf9
1 parent b4a536e
commit ad69bf9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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: