Fix variable unpacking

WildMeOrg · Aug 15, 2024 · a8ae4fa · a8ae4fa
1 parent 7fa72ec
commit a8ae4fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wbia_miew_id/train.py b/wbia_miew_id/train.py
@@ -46,7 +46,7 @@ def run_fn(self, model, train_loader, valid_loader, criterion, optimizer, schedu
                 print('Group average score: ', valid_score)
             else:
                 print('Evaluating on full test set')
-                valid_score = eval_fn(valid_loader, model, device, use_wandb=use_wandb)
+                valid_score, valid_cmc = eval_fn(valid_loader, model, device, use_wandb=use_wandb)
                 print('Valid score: ', valid_score)
 
             if valid_score > best_score: